在远程服务器上运行代码时出现AssertionError: Torch not compiled with CUDA enabled

问题分析

这个问题常常是由于没有安装pytorch对应的GPU版本导致的。

解决方法

我安装的是pytorch=1.7.1，使用conda search pytorch命令搜索anaconda上可用的pytorch版本，

从图中可以发现，pytorch=1.7.1的只有cpu版本，这个问题可以通过换源来解决，通过Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror找到对应你系统上的源，

由于我的是linux系统，所以点击linux-64,然后根据自己的python版本和对应的CUDA版本，选择适合自己的包。

这里我的python版本是3.7，选择的cuda=10.1,使用命令

conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
成功安装。