有3张显卡但是torch.cuda.device_count = 1 用nvidia-smi查看GPU信息如下显示 但是在使用时,使用torch.rand(3,3).cuda('1)或者torch.rand(3,3).cuda('0‘’)出现错误CUDA error : invalid device ordinal,使用torch.cuda.device_count()查询,结果为1,如下图所示 后来通过检查环境变量,发现环境变量设置有问题,原始设置如下所示,系统默认了GPU显卡为第二块显卡,所以出现了以上错误 通过修改的此处环境变量为如下所示,问题解决