发布时间:2023-04-20 文章分类:电脑百科 投稿人:李佳 字号: 默认 | | 超大 打印

print(torch.cuda.is_available()) 返回false的解决办法

1.问题简述
今天给新电脑配置pytorch深度学习环境,最后调用python打印print(torch.cuda.is_available())一直出现false的情况(也就是说无法使用GPU),最后上网查找资料得出报错的原因:下载的pytorch是CPU版本,而非GPU版本。
print(torch.cuda.is_available()) 返回false的解决办法

2.报错原因
按照最开始的方法,在pytorch的官网上根据自己的cuda版本(笔者为cuda11.5)使用对应的指令在conda prompt中在线下载:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch。为了使下载速度加快,我们一般会将后面的-c python删除,采用清华镜像源地址进行安装,例如:

conda install pytorch torchvision torchaudio cudatoolkit=11.3

(清华源加速需要提前部署好,可参考其他博客)。
print(torch.cuda.is_available()) 返回false的解决办法

然鹅错误就出在清华镜像源这里,从网上查资料可得,镜像源下载的一般为CPU版本,也是这导致前面所说一直返回false。如果不使用清华源,在官网下载速度又太慢,这个时候我们考虑pip安装。

3.解决办法
首先进入离线文件的下载地址:link.
print(torch.cuda.is_available()) 返回false的解决办法
在其中找到符合自己的那个文件。注意看清要下载的包的名字,torch还是其他啥。cuxxx一般是支持cuda的包。看好自己环境中的python处理器是什么版本的,不要下载错了,比如python3.6就是cp36。window,linux版本区分清楚。如下图所示就是python对应版本出错后的结果(后续安装离线文件报错)
print(torch.cuda.is_available()) 返回false的解决办法
下载完离线文件后,将离线文件放在你需要安装的环境中,如下图,我放在了我自己的深度学习环境python36中,在anconda的envs中。

print(torch.cuda.is_available()) 返回false的解决办法
最后一步就是打开conda prompt ,从base环境切到你自己的环境里,通过cd命令切换到那个环境的文件地址中(切cd操作可以参考其他文章),同时最后输入指令pip install 文件名,例如:pip install torch-1.10.2+cu113-cp36-cp36m-win_amd64.whl
print(torch.cuda.is_available()) 返回false的解决办法
最后静等不到一分钟,就安装好了。去python测试,终于不报错了,也能显示存在一张工作的显卡
print(torch.cuda.is_available()) 返回false的解决办法
检查GPU是否可用的代码:

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())

最后结果:
print(torch.cuda.is_available()) 返回false的解决办法
这里有一个报错是numpy的版本不对,后面卸载了重装就是了。