报错:
我的电脑没有英伟达显卡,装不了cuda,只能用CPU跑神经网络,这个报错是说Torch找不到cuda ,无法编译。
解决办法:
在代码开头加上
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
然后将其余的.cuda()改成.to(device),torch就可以在CPU环境运行了。
多说一点:
如果是有英伟达显卡且电脑不止一张显卡,在代码开头写
device = torch.device('cuda:0')
就是选择第一个显卡作为GPU环境