解决pytorch检测不到cuda的问题/pytorch找不到GPU

最新推荐文章于 2024-04-24 13:30:10 发布

薛猫颚的腚

最新推荐文章于 2024-04-24 13:30:10 发布

阅读量2.5w

点赞数 7

文章标签：深度学习 pytorch python

本文链接：https://blog.csdn.net/weixin_44952817/article/details/123294673

版权

检测代码：

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
#返回cuda表示成功
#或者
print(torch.cuda.is_available())
#返回True表示成功

用这两条条代码可以检测是否能找到GPU，如果找不到可能有如下几个原因：

1.没有安装CUDA/CUDNN

安装CUDA网址：https://developer.nvidia.cn/cuda-downloads
安装CUDNN：https://developer.nvidia.cn/rdp/cudnn-archive
记得CUDA要配准环境
注意1：如果直接搜索CUDA/CUDNN网址打不开是因为使用了.com的网址，用.cn就可以正常访问了
注意2：CUDA和CUDNN的版本一定要对应好，截止2022年3月建议安装11.3之前的版本，因为最新的版本可能跟pytorch还没有适配

2. CUDA和pytorch的版本无法匹配

这条原因最常见，CUDA和pytorch的对应版本怎么查找呢？
到这个网址：https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
这里的CU113指的是CUDA11.3版本，torch-1.10.1指的是torch版本，cp-36指的是python 3.6版本，linux，win这些就不用我解释了吧。
那whl文件的使用呢？
我们将它下载到桌面，用WIN+R打开cmd，然后进入到桌面，利用pip安装即可
在这里插入图片描述