Script running tool for optimizing GPU memory utilization.
Project description
TRIAGE
Intended use: running a multitude of GPU-intensive scripts in a way that optimizes GPU memory utilization. Great for ML/DL based experiments on servers shared between several users.
Installation
pip install triage-runner
Usage
See --help
option for extended list of possible arguments.
Running one config:
triage run_config.json
Running several configs:
triage run_config1.json run_config2.json run_config3.json
Patterns can be used for config discovery as well:
triage run_config*.json
More on pattern syntax can be found here: https://docs.python.org/3.10/library/pathlib.html#pathlib.Path.glob
Run configurations
Stored in JSON format. The sample run configuration looks like this:
{
"memory_needed": 10.0,
"config_name": "sample_config",
"command": "python3 train.py",
"args": [
"arg1",
"--arg2",
["--seed=1", "--seed=2", "--seed=3"],
"--arg3=3"
]
}
Every entry in args
list is an argument for command
. An entry can be a list - in which case TRIAGE will iterate through all the possible combinations of all values in list entries. The example script above will be run 3 times with an argument --seed
set to 1, 2 and 3.
Parameter config_name
is optional and is used for logging the results (see --logfile
option). Based on this parameter environment variable TASK_NAME
is set in order to be used by running script.
Project details
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
Hashes for triage_runner-0.0.17-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0c05fe4fa86bee34ec10f2700e398fe19e42902bb692adb14315b510c870151e |
|
MD5 | 1b47d329894fe352df10f63ee958e37c |
|
BLAKE2b-256 | 96e7b5e1ecbbd9ffc7f8ba3dfeecdef4df1ea098d671b65c3267f2bab3a32c1e |