PyTorch报错：AssertionError: Torch not compiled with CUDA enabled

发使用pip 安装pytorch会超时，安装失败，使用cuda安装就会很快

我的cuda版本是11.2，没有找到与之对应的版本

 就使用cuda=10.2了（python==3.6.2）

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

然后在pycharm中运行程序，报错

AssertionError: Torch not compiled with CUDA enabled

此时程序中配置信息是：

torch.cuda.set_device(-1)
net = Net_SC().cuda()
loss_func = nn.CrossEntropyLoss().cuda()

感觉是GPU和CPU版本的原因

在新的环境中安装CPU的pytorch，(仍然是python==3.6.2）

conda install pytorch torchvision torchaudio cpuonly

并在程序开头加入

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 

 

并把程序中所有的 .cuda()改为.to(device)

程序正常运行