提交 badfab33 编写于 作者: M MRXLT

add check

上级 1e9af5e8
......@@ -69,7 +69,8 @@ def start_multi_card(args): # pylint: disable=doc-string-missing
gpus = args.gpu_ids.split(",")
if "CUDA_VISIBLE_DEVICES" in os.environ:
env_gpus = os.environ["CUDA_VISIBLE_DEVICES"].split(",")
if int(gpus[-1]) >= len(env_gpus):
for ids in gpus:
if int(ids) >= len(env_gpus):
print(
" Max index of gpu_ids out of range, the number of CUDA_VISIBLE_DEVICES is {}.".
format(len(env_gpus)))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册