Ubuntu-torch.cuda.is_available() 的返回值为False

Ubuntu-torch.cuda.is_available() 的返回值为False

按照流程安装了GPU版本的pytorch,但是输入测试代码:

>>>import torch
>>>torch.cuda.is_available()

结果一直返回结果为False,如果提示是这个则代表GPU版本的pytorch无法正常使用。
结合网上其他的原因,检查了一遍,发现我的cuda和cudnn都安装成功了,版本也应该是匹配的。无奈只能回归到官网去寻找解决方案了

官网提示


注意到下面有对应的提示:先决条件
请按照此处描述的步骤进行操作:: https://github.com/pytorch/pytorch#from-source.

在这里插入图片描述
这里提示要先安装完CUDA和cudnn。接着安装依赖项,因为我是ubuntu18.04的版本,所以直接在终端创建好的虚拟环境下敲入代码:

conda install -c pytorch magma-cuda101#我之前安装的CUDA是10.1的版本

如下图所示(可以看到之前提示False的地方):
在这里插入图片描述
那么等待安装完之后再测试一下,结果就是True了!

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值