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

检测代码:

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安装即可

来源:薛猫颚的腚

物联沃分享整理
物联沃-IOTWORD物联网 » 解决pytorch检测不到cuda的问题/pytorch找不到GPU

发表评论