File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -127,8 +127,8 @@ def prepare_nvidia_docker_run(args):
127127 raise UserDockerException (
128128 'ERROR: GPU %d is currently not available!\n Use:\n '
129129 '"sudo userdocker ps --gpu-free" to find available GPUs.\n '
130- '"sudo userdocker ps --gpu-used" or "nvidia-smi" to see '
131- 'status information .' % g
130+ '"sudo userdocker ps --gpu-used" and "nvidia-smi" to see '
131+ 'status.' % g
132132 )
133133 else :
134134 # NV_GPU wasn't set, use admin defaults, tell user
@@ -141,7 +141,9 @@ def prepare_nvidia_docker_run(args):
141141 gpus = gpus_available [:gpu_default ]
142142 if len (gpus ) < gpu_default :
143143 raise UserDockerException (
144- "Could not find %d available GPU(s)" % gpu_default
144+ 'Could not find %d available GPU(s)!\n Use:\n '
145+ '"sudo userdocker ps --gpu-used" and "nvidia-smi" to see '
146+ 'status.' % gpu_default
145147 )
146148 gpu_env = "," .join ([str (g ) for g in gpus ])
147149 logger .info ("Setting NV_GPU=%s" % gpu_env )
You can’t perform that action at this time.
0 commit comments