CUDA error: invalid device ordinal 解决方法

出现该问题通常是由于显卡设备调用问题，包括但不限于：
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的编号即可