Pytorch错误:Torch not compiled with CUDA enabled

  在用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)如果没有可用的显卡资源,则检查代码,把调用显卡资源的部分全部去掉即可

你可能感兴趣的:(Pytorch打怪之路,cuda,python,pytorch,debug,bug)