解决问题：AssertionError: Torch not compiled with CUDA enabled

报错：

我的电脑没有英伟达显卡，装不了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环境