在用Pytorch炼丹的过程中，很意外地碰到了一个错误提示AssertionError: Torch not compiled with CUDA enabled，如下图所示。

错误提示   代码本身可以确定是没有问题的，因为只是在往日已经跑过的代码上加入了小改动。最后发现问题出现在Pytorch和CUDA版本不兼容的问题上。问题的发现可以在终端中输入'python'命令，运行python后，输入

import torch
print(torch.__version__)
print(torch.cuda.is_available())
复制

  如果输出的结果是False，那么说明当前的Pytorch版本无法使用显卡。

解决办法：

  1）如果有显卡资源并需要使用显卡资源，先检查一下当前的CUDA版本，在终端中输入

cat /usr/local/cuda/version.txt
复制

然后根据自己的CUDA版本在Pytorch官网查看并安装可用的版本，参考链接如下:

CUDA 10.2

https://download.pytorch.org/whl/cu102/torch_stable.html

CUDA 10.1

https://download.pytorch.org/whl/cu101/torch_stable.html

CUDA 10.0

https://download.pytorch.org/whl/cu100/torch_stable.html

CUDA 9.0

https://download.pytorch.org/whl/cu90/torch_stable.html

CPU及所有CUDA版本

https://download.pytorch.org/whl/torch_stable.html

  2）如果没有可用的显卡资源，则检查代码，把调用显卡资源的部分全部去掉即可