五步解决torch.cuda.is_available()返回False的问题

First of all,感谢博主:dysljsjsq 链接

基于上次发的帖子,我是用Anaconda创建虚拟环境装的pytorch。

可以看到我在官网上选择的是Stable(1.4),cuda10.1,linux,conda安装

conda install pytorch torchvision cudatoolkit=10.1 (我把-c pytorch去掉了,这样的方式安装快)
五步解决torch.cuda.is_available()返回False的问题_第1张图片然后,我用了一段时间,import torch完全没问题,可以创建tensor类型,完全ok。但我一用到和cuda有关的语句,就报错,说不存在cuda。我一开始不懂为什么,查了很久之后,也尝试了很多博主说的驱动版本可能不对,更新版本之类的,重装过Nvidia。
五步解决torch.cuda.is_available()返回False的问题_第2张图片驱动版本和cuda版本完全没问题。顺便附上版本对应表
五步解决torch.cuda.is_available()返回False的问题_第3张图片
最终查到开头那个博主的教程,第六感告诉我,我可能是这种情况!!!然后就查了我pytorch的版本

勘探错误

python
import torch
print(torch.__version__)

如下图。是1.3.1!!!!!记得我在官网上查到的版本是什么吗????是1.4啊啊啊啊啊!!所以,ok大概找到问题所在了。

开始解决

1.卸载pytorch
pip uninstall torch
2.尝试用官网pip方法重新安装,ok,fine,我不配,好慢好慢,我等了又等,结果就是丢包严重,下不下来。
五步解决torch.cuda.is_available()返回False的问题_第4张图片3.手动下载安装包
注意文件名是对应的版本号!!
例如:cu101是cuda10.1
五步解决torch.cuda.is_available()返回False的问题_第5张图片要下载两个,一个是torch框架,一个是torchvision库

4.在安装包位置的文件夹下打开终端,分别安装

pip install torch-1.4.0-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.2.0-py2.py3-none-any.whl

5.之后再尝试一下就能得到ture了!!!终于不是CPU假装GPU运行的用户了!!
五步解决torch.cuda.is_available()返回False的问题_第6张图片

最后,我也不知道为啥conda不能安装GPU版本的torch,望大神指教。

你可能感兴趣的:(怡宝升级记,cuda,gpu,pip,深度学习,pytorch)