栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/754754.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号