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资源,但是正式提交平台后单卡训练

Logo

鸿蒙生态一站式服务平台。

更多推荐