ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-04-24 16:14:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move --gpus and --cpu-only to worker args (used by daemon, execute and build)

Browse Source
This commit is contained in:
allegroai 2020-05-09 19:51:45 +03:00
parent 5bb257c46c
commit abb30ac2b8
1 changed files with 11 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
trains_agent/interface/worker.py
View File

@ -30,6 +30,17 @@ WORKER_ARGS = {
'type': lambda x: x.upper(), 'type': lambda x: x.upper(),
'default': 'INFO', 'default': 'INFO',
}, },
'--gpus': {
'help': 'Specify active GPUs for the daemon to use (docker / virtual environment), '
'Equivalent to setting NVIDIA_VISIBLE_DEVICES '
'Examples: --gpus 0 or --gpu 0,1,2 or --gpus all',
'group': 'Docker support',
},
'--cpu-only': {
'help': 'Disable GPU access for the daemon, only use CPU in either docker or virtual environment',
'action': 'store_true',
'group': 'Docker support',
},
} }
DAEMON_ARGS = dict({ DAEMON_ARGS = dict({
@ -45,17 +56,6 @@ DAEMON_ARGS = dict({
'default': False, 'default': False,
'group': 'Docker support', 'group': 'Docker support',
}, },
'--gpus': {
'help': 'Specify active GPUs for the daemon to use (docker / virtual environment), '
'Equivalent to setting NVIDIA_VISIBLE_DEVICES '
'Examples: --gpus 0 or --gpu 0,1,2 or --gpus all',
'group': 'Docker support',
},
'--cpu-only': {
'help': 'Disable GPU access for the daemon, only use CPU in either docker or virtual environment',
'action': 'store_true',
'group': 'Docker support',
},
'--force-current-version': { '--force-current-version': {
'help': 'Force trains-agent to use the current trains-agent version when running in the docker', 'help': 'Force trains-agent to use the current trains-agent version when running in the docker',
'action': 'store_true', 'action': 'store_true',
@ -153,15 +153,6 @@ COMMANDS = {
'nargs': '*', 'nargs': '*',
'default': False, 'default': False,
}, },
'--gpus': {
'help': 'Specify active GPUs for the docker to use'
'Equivalent to setting NVIDIA_VISIBLE_DEVICES '
'Examples: --gpus 0 or --gpu 0,1,2 or --gpus all',
},
'--cpu-only': {
'help': 'Disable GPU access (cpu only) for the docker',
'action': 'store_true',
},
'--python-version': { '--python-version': {
'help': 'Virtual environment python version to use', 'help': 'Virtual environment python version to use',
}, },