Backend implementation for running MLFlow projects on Slurm
Project description
MLFlow-Slurm
Backend for executing MLFlow projects on Slurm batch system
Usage
Install this package in the environment from which you will be submitting jobs. If you are submitting jobs from inside jobs, make sure you have this package listed in your conda or pip environment.
Just list this as your --backend in the job run. You should include a json
config file to control how the batch script is constructed:
mlflow run --backend slurm \
--backend-config slurm_config.json \
examples/sklearn_elasticnet_wine
It will generate a batch script named after the job id and submit it via the
Slurm sbatch command. It will tag the run with the Slurm JobID
Configure Jobs
You can set values in a json file to control job submission. The supported properties in this file are:
| Config File Setting | Use |
|---|---|
| partition | Which Slurm partition should the job run in? |
| account | What account name to run under |
| gpus_per_node | On GPU partitions how many GPUs to allocate per node |
| gres | SLURM Generic RESources requests |
| mem | Amount of memory to allocate to CPU jobs |
| modules | List of modules to load before starting job |
| time | Max CPU time job may run |
| sbatch-script-file | Name of batch file to be produced. Leave blank to have service generate a script file name based on the run ID |
Development
The slurm docker deployment is handy for testing and development. You can start up a slurm environment with the included docker-compose file
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file mlflow_slurm-1.0.0.tar.gz.
File metadata
- Download URL: mlflow_slurm-1.0.0.tar.gz
- Upload date:
- Size: 6.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.8.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 32cc72abbb3f4e462ec1a4937a96dc6e64cb4e6e58ee45c3f21544f8c5c5beb8 |
|
| MD5 | c2270ef54bf74a0a7b3e00badb08d841 |
|
| BLAKE2b-256 | 2e6c904eacbdb5a1535ad1246d8de2e15bc9d0ef7531e63486487ebc2f57092f |
Provenance
File details
Details for the file mlflow_slurm-1.0.0-py3-none-any.whl.
File metadata
- Download URL: mlflow_slurm-1.0.0-py3-none-any.whl
- Upload date:
- Size: 7.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.8.18
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 071c2a307b4985aa535387b2cdb6309ce10654906029047fa91a4d94e7ce3b00 |
|
| MD5 | 17ac4a7f5f654218b4b37effba7cd984 |
|
| BLAKE2b-256 | 17b491d17c176252ddbae4d0de7cb897bd48bd978e4c95949ff7344232ffbc23 |
