解决torch.cuda.is_available()返回结果为False

装了好几天的环境,死活用不了GPU
torch.cuda.is_available()返回结果一遍一遍的都是False,心态都崩了!!!
一开始以为是nvidia驱动的问题,然后看了几篇帖子,总结一下我的错误

一、cuda版本问题
cuda是向下兼容的,一开始我用的cuda10.1的docker镜像,本机是cuda10.0的,所以返回False。要确保新建的环境的cuda版本要比本地的基础环境低!!!

二、pytorch和cuda版本对应问题
在这里插入图片描述
装的torch==1.4.0版本的,用print(torch.__version__)查了一下我装的这个torch是要求cuda是10.1版本的。

问题来了:
pytorch官网上和pip install这两种方法的安装方式最大的缺点就是安装的pytorch指定的cuda版本不可自己控制,所以就会出现上图的版本不对应的情况。

那么我想要一个torch==1.4.0和cuda9.2的环境,就需要下载这两个相对应的版本

于是乎我在这个网站 各种torch、torchvision 找到了满意的torch

  • 32
    点赞
  • 149
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值