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.15-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0572fde9f1fcbb53ebca60968e8d2958e4ca76ca5c79afd26626e9eed6e317fe |
|
MD5 | b915dcee464f478b6c8c3a712e1240c8 |
|
BLAKE2b-256 | 8e1a5181eca085204f3122d53165e591c8fef869e3858d1964f5d83560b903fc |