欢迎光临散文网 会员登陆 & 注册

Pytorch(GPU版)在命令窗口安装成功后仍无法使用GPU的解决方法

2023-07-25 22:01 作者:阿宅学习中  | 我要投稿

依据土堆老师的环境配置教程【最详细的 Windows 下 PyTorch 入门深度学习环境安装与配置 CPU GPU 版 | 土堆教程】进行环境配置,一切都很顺利,直到下载pytorch时,不论是直接官网下载还是镜像通道,最后的torch.cuda.is_available()返回值都是false

上网找了不少文章和官网之后摸索出了解决方法,故分享。

【注意】本文适用于【能够从镜像网站中顺利下载pytorch的各种包】,并且【最后的torch.cuda.is_available()返回值是false】的朋友们。


【出现问题的原因】

请参照【https://windses.blog.csdn.net/article/details/125910538】这位老师的文章,说得非常详细易懂,本文的解决方法也是完全参照老师的步骤。

(老师的文章中给出的是linux的解决方法,不能够直接用于windows,为了帮各位有同样问题的朋友快速解决问题而不像我一样整了一天后还继续摸索,于是写了这篇文章)

老师总结出来的根本原因是:【找不到对应GPU版本的pytorch文件,所以conda就用CPU替代了=v=】

我们下载的几个包都有不同的版本,镜像站中没有恰好对应这几个包这几个版本组合起来的pytorch(GPU版),就自行用pytorch(CPU版)代替了。而且我们检查的时候,按照土堆老师的教程,检查到了【pytorch】就可以继续进行下一步检查,但这个pytorch是CPU版本的。

所以torch.cuda.is_available()返回值一直都是false。

所以我们的解决方法就是:找到镜像站搭配好的、有pytorch能够对应的组合。


【解决方法】

1.首先按照土堆老师的教程,选择好自己电脑能够能够使用的CUDA版本

2.在选择的镜像站中搜索自己想要搭建的虚拟环境的python版本,举例如Microsoft Edge用【ctrl+F】便可在网页内搜索。


【例:】

在土堆老师给出的镜像站中选择【下载pytorch, torchvision, torchaudio】的清华镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/


下载pytorch, torchvision, torchaudio的清华镜像站

用浏览器内置搜索功能在网页中搜索我们的虚拟环境所用的python版本,并且找到自己刚刚选择的CUDA版本

注:edge浏览器按【ctrl+F】即可打开网页搜索功能

以python3.8和CUDA=11.3为例,进行查找,查找方式和结果如图。

python3.8和CUDA11.3为例,查找结果

我们发现,提供的pytorch版本为1.10.2,有这个信息已经足够了。


3.进入命令窗口,在镜像通道进行下载。

打开anaconda prompt,进入自己创建的虚拟环境。

进入虚拟环境

输入conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/  #从镜像网站中下载cuda=11.3

输入conda install pytorch==1.10.2 torchvision torchaudio -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/  #从镜像网站中下载 pytorch==1.10.2和剩余的几个包

剩余的包没有版本要求,只有pytorch和cuda有版本要求。

pytorch的版本确定已在上一步提过。这个非常重要。

看系统给出的包,观察pytorch这一项的内容,里边所含的是否和镜像站所含的各种包【pytorch==1.10.2 python==3.8 cuda==11.3 cudnn==8_0】的版本是一致的(就是上上图中红色圈出来的那个)

确认完全一致之后继续,安装完成之后再按照土堆老师的教程检验,得到的结果就是True了。





Pytorch(GPU版)在命令窗口安装成功后仍无法使用GPU的解决方法的评论 (共 条)

分享到微博请遵守国家法律