本人用了一天时间才整明白的道理
一. 对于电脑有显卡的铁子
在终端输入
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果输出的结果是False,那么说明
1.当前的Pytorch版本无法使用显卡,即下载的pytorch版本是cpu版本
2.cuda未安装或者版本不对口哦。一般版本不对会提示版本问题的,所以大部分人都是第一个问题。
二. 对于电脑有没有显卡的 铁子
由于部分没有显卡的电脑是无法安装驱动nvidia的,所以没有cuda,此时你只需要改动以下代码
1.在主程序前面加上
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
2.将.cuda()改成.to(device)即可
如果这还不行,我把桌子吃了