参考的大神连接: 安装pytorch报错torch.cuda.is_available()=false的解决方法
巨坑之默认的安装源为清华源
原因说明
之前因为conda本身的安装源速度并不是很快,故将默认的安装源换成了清华源,本身也觉得没有任何问题,但是在安装pytorch的时候出现了难以发现的错误。
如下图:从pytorch的官网下载自己所需要的版本,然后复制了最后一行的下载命令,
下载的时候因为默认源为清华源,很顺畅,得劲。下载完,进行测试的时候,怎么都不行。
测试代码为(linux 下的代码):
先激活刚刚下载pytorch对应的虚拟环境
conda activate pytorch
在进入python环境
python
导入pytorch包
import torch
验证是否可以使用cuda
torch.cuda.is_available()
如果你安装的一切都是正确的话,返回的结果应该为:
True
但是我一直为False,查看了无数次的cuda和pytorch对应的版本,也重启了无数次,直到我使用
conda list
查看我的安装包的时候,我突然注意到我安装的pytorch后面的注释为py37_cpu_0,如下,我就很纳闷了,我安装的不是cuda版本吗,然后又是无数次的卸载安装,还是这个版本,好无奈
后面查看到一篇大神博客,醍醐灌顶,大神连接:
好像是清华源没有pytorch对应的cuda,所以一直给你下载的都是cpu版本。
解决办法
方法一(较为方便)
将conda下载源换回为默认源或者修改默认源为科大源:
conda config --remove-key channels
切换之后,清华源将不再起作用,起作用的为默认源。
使用下载命令(这个对应自己在pytorch上所给的下载连接)
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
但是在此情况下,下载容易中断,所以建议切换到pip下载好一点(将conda改成pip)如下:
复制下面连接下载:
pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
我就是这样解决了问题。希望大家不再被这个问题折磨。
方法二
手动去pytorch官网去下载上述自己所需的安装包,然后手动进行安装。具体可参考大神的博客
文章出处登录后可见!
已经登录?立即刷新