pytorch的问题:解决引用torch,但torch.cuda.is_available() 返回false

我在网上找了超多方法,说一下我这次配的经验吧,也是以防自己忘了。
pycharm、python、Anaconda3、cuda等我之前配过,稀里糊涂配的,大家网上搜搜很多博客,所以这里就不提了,主要是我也不记得了。(●ˇ∀ˇ●)
第一次配环境,如果以下步骤有不对的地方,也请各位大佬指正

步入正题:

我是win10系统

  • ①检查电脑NVIDIA的cuda版本
    从NVIDIA控制面板确定电脑NVIDIA的cuda版本:NVIDIA控制面板→帮助→系统信息→组件→查看文件名为NVCUDA.dll 的产品名称NVIDIA CUDA ****** driver (******的部分就是电脑NVIDIA的cuda版本)
    我的是10.2.89,后面就以cuda10.2.89为例,自己对应自己的版本
  • ②检查自己电脑的nvidia驱动是否能带的起cuda 10.2
    进去cmd(命令提示符,win+R输入cmd就进去了) 输入nvidia-smi(如果不出来的去网上搜下怎么打开nvidia-smi)
`nvidia-smi`

如图划红线的地方所示(提醒NVIDIA-SMI与Driver Version是要一样的哟,这就是显卡驱动版本号)
在这里插入图片描述

显卡驱动版本号与CUDA版本号对应关系
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html在这里插入图片描述
从图可看到我的是446.14 ,而CUDA10.2.89的要求是≥441.22 。
所以我电脑的显卡驱动是允许装CUDA 10.2的。

  • ③检查环境中已安装cuda的版本
    依旧是在cmd(命令提示符)内输入nvcc -V (注意第二个V是大写)
nvcc -V

在这里插入图片描述
如图所示我的是10.2,所以配的是对的,如果你的和第①步看的CUDA版本不一样,那就要重装对应版本的CUDA驱动程序了(最好是对着第①步的CUDA版本配,当然你也可以试试更改第①步NVIDIA显卡驱动的CUDA版本,反正要想成功配pytorch 以上提到的关于CUDA的版本一定要“对齐”
- 想找与显卡同版本的cuda驱动程序,官方下载链接是这个https://developer.nvidia.com/rdp/cudnn-download(需要登陆/注册才能下载,注册很简单。)
- 安装对应版本的CUDA驱动程序可以参考下这篇文章https://blog.csdn.net/u011473714/article/details/95042856

可以通过命令conda list 查看自己配的包,如图所示
在这里插入图片描述
如果conda list没有找到cudatookit和cudnn,可以再弄一下conda install cudatookit=10.2就可以,然后conda install cudnn

但是

conda安装pytorch国外的镜像容易报错安装不上,得配国内的镜像,有很多文章教。但是我觉得镜像麻烦,明明可以直接pip配为什么这么多人推荐的方法都是改镜像,如图画篮圈的地方选了pip,然后复制划红线的内容安装pytorch即可。
在这里插入图片描述
然后conda list 查看
在这里插入图片描述
我的是这几个都与CUDA版本“对齐”就torch.cuda.is_available()返回true了。奈斯!

  • 8
    点赞
  • 43
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值