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

评论 26
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值