出现该问题通常是由于显卡设备调用问题,包括但不限于:
1. 超出显卡编号/不存在该显卡,解决方案:使用nvidia-smi确定显卡编号。
2.错误的调用/torch未正确安装 :
import torch
x = torch.rand(5, 5)
x = x.cuda() # 尝试将 tensor 移动到 GPU
print(x)
通过上述测试代码查看GPU能否正常调用。
3.通过前列排查确定环境/GPU/torch版本正常:则可以通过以下代码来进行device的重新传值解决问题:
device = torch.device('cuda:1')
其中 cuda:1更改为自己所需GPU的编号即可