RuntimeError: No CUDA GPUs are available问题解决

本地在调试机上跑没问题，但是上传到服务器上正是跑出现问题，出现如下报错

1.查看是否具有GPU

import torch
print(orch.cuda.is_available())
print(torch.cuda.device_count())

如果均显示正常，可以查看代码中是否有：

os.environ['CUDA_VISIBLE_DEVICES']='3'

修改为“0”，”1“ ，问题解决！
原因：
调试机上有4块GPU资源，但是正式提交平台后单卡训练