Torch not compiled with CUDA enabled

本人用了一天时间才整明白的道理

一. 对于电脑有显卡的铁子

在终端输入

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)即可

如果这还不行，我把桌子吃了