明明安装了gpu版的pytorch可是测试的时候居然用的试cpu版的?

在这里插入图片描述

但我安装的时候明明安装的cuda对应的pytorch

用的清华镜像
命令是conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

然后用测试指令在python下进行测试

torch.cuda.is_available()

返回false说明gpu不能用
网上普遍的问题有三种
(1)你的CUDA,cuDNN,torch版本是否对应

(2)清华镜像下下载的torch
我的属于第二种情况
这种情况不是咱们安装步骤的问题而是conda镜像的问题,
你以为下载的是GPU版本,其实镜像下载的是cpu版本你必须手动去官网用pip下载安装

解决方法
先卸载,pip uninstall torch
我当时用了这个命令,然后又到了anaconda我所安装pytorch的环境目录下又检查还有没有残余

D:\anaconda\envs\(环境名称)\Lib\site-packages

对照着下图如果有一样的删除
在这里插入图片描述

正确下载方法

去pytorch官网

因为当时显示我cuda10.2的对应的pytorch windows pip的没有了,我就直接自行找的对应的压缩包下载了

这里面关键的思路就是,你自己手动去官网下载你要找的压缩包,下载,然后你用anaconda prompt 进入环境,然后进入你下载的压缩包的文件路径(cd d:\xx\xx),然后输入 pip install torch==压缩包的全名,就能安装上了
直接进入这个网址然后找你所需对应的东西,要耐心细心

https://download.pytorch.org/whl/torch_stable.html
因为我之前参考了另一篇博客
就下载了这三个
我没装torchaudio,我暂时用不到音频,主要是我不知道载哪个版本的。。。
在这里插入图片描述
具体命令与测试请见下图,我成功了
在这里插入图片描述

至于下载过程中的时间,你要嫌慢用那个(vpn)
感谢这篇博客给我了很大启发,并且第一个问题在这篇博客上也写了出来

https://blog.csdn.net/qq_46126258/article/details/112708781

  • 3
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值