RuntimeError: CUDA error: invalid device ordinal
问题描述：
这个是什么原因呢，程序在pycharm上运行，使用的是pytorch框架，然后报错了。
在网上查询时都说是gpu被占用，但是我不知道应该怎么改，
关于device的代码如下
关于gpu的代码如下
怎么改？？求解
答
在程序的前面加上，后面的数字要看你的显卡数目，意思是对该程序显示哪几张显卡可以使用。如果只有一张的话，要改成0.然后就是torch设置显卡的问题。
最好这么写：torch.device('cuda:0')
import os
os.environ['CUDA_VISIBLE_DEVICES']='0,1,2,3' #4路显卡
答
建议这么写： net.cuda()