踩坑系列之pytorch安装之后不能使用cuda


参考的大神连接: 安装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官网去下载上述自己所需的安装包,然后手动进行安装。具体可参考大神的博客

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年3月7日 下午10:46
下一篇 2023年3月7日 下午10:49

相关推荐