Skip to content

Commit b322279

Browse files
committed
also improve none NV_GPU no available GPU error message
1 parent 9836264 commit b322279

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

userdocker/subcommands/run.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,8 @@ def prepare_nvidia_docker_run(args):
127127
raise UserDockerException(
128128
'ERROR: GPU %d is currently not available!\nUse:\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)!\nUse:\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)

0 commit comments

Comments
 (0)