torch.cuda.is_available()=false 完美解决方法

torch.cuda.is_available()=false 完美解决方法

首先看你电脑的驱动是不是安装好了,另外看驱动和cuda匹配不;

你安装cuda的文件包名字都是带有一个驱动版本的,最低是那个版本,大于这个版本的都可以

比如:cuda_10.1.243_426.00_win10.exe,这个驱动就是426.00

第二,看你的cuda和cudnn安装是否成功以及路径问题,以下是两个安装地址,注意版本匹配

CUDA Toolkit Archive | NVIDIA Developer

https://developer.nvidia.com/rdp/cudnn-download

安装后使用cmd的nvcc -V自查

第三,看你的pytorch,torchvision等版本是否一致,同时也要看你的torch是不是cpu版本

使用conda list看,如果显示版本是什么pyxx_cpu,那就是cpu版本,正确的应该是 pyxx_cuxxx

这个一定要看,因为你在官网也是会下载到cpu版本的,比如官网上的代码conda xxx -c pytorch,当你有依赖的库的情况下,是有机会给你下cpu版本的。

最后打开cmd的python
(dete2for_python3.6) D:\sbj>python
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] 
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

这样就可以了

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值