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版本的。

扫描二维码关注公众号,回复: 17520257 查看本文章

最后打开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

这样就可以了

猜你喜欢

转载自blog.csdn.net/bieliwuguiqi/article/details/124840518