conda安装GPU版pytorch,结果是cpu版本

conda环境安装pytorch

跟随土堆大佬安装pytorch学习
https://www.bilibili.com/video/BV1hE411t7RN/?p=1&spm_id_from=pageDriver&vd_source=ab6eed2b82a124efe2304bcb92eab646

结果conda安装GPU版pytorch,结果却是cpu版本

总结一下原因:
1.你选择的cuda版本一定要低于你的实际支持的版本,不然不兼容,血泪教训
在这里插入图片描述
我这里最高支持11.3
安装的就是10.2
在这里插入图片描述

Windows 下的 CUDA 10.2 选择 pip 安装稳定版,命令为

pip3 install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

如果上面这条命令下载速度太慢,可以用 -i 临时使用清华镜像下载

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.

2.安装的三个版本在源中没有
[python|cudatoolkit|pytorch]这三个版本不同的组合,真的不一定有。
三个版本组合一定要对
具体参考
https://blog.csdn.net/u013468614/article/details/125910538

>>> import torch
>>> torch.cuda.is_available()
true

终于搞完了

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值