Anaconda 中安装pytorch GPU 版本(踩坑记录)

1. 安装pytorch踩坑记录

网上完整步骤的安装贴已经有不少,说下重点解决的,直接使用官网指令安装pytorch11.3发现安装的是CPU版,GPU不能用,含泪找解决步骤。
网上有人说用查看有无cpuonly的方法,试过不管用,具体参见链接安装GPU版本pytorch

解决办法是卸载相关pytorch安装文件,重新安装,使用指令

pip uninstall torch

发现只能卸载由pip安装的torch, 原来由conda安装的依然在,当然也可以搜**使用pip卸载pytorch,很多条指令,逐个库下卸载*,不要觉得没必要,因为直接一条指令卸载卸不干净,已经试过。

还是要老老实实的用conda卸载,时间巨久,我也就是花了十几个小时吧。。。

conda uninstall pytorch
conda uninstall libtorch

这次发现用conda卸载了一堆依赖库,嗯,这次大概可能已经卸载干净了吧?
老老实实的用whl自己下载安装包下载,参考安装GPU版本pytorch,今天网络太差了,pytorch官网登不上,等会再继续写。
换了清华源,参考博客pytorch加速下载——清华镜像源,因为我早就改过,具体步骤就不说了,重点是添加临时下载源,

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

接下来,我的cuda是11.3,所以就安装相应GPU版本的pytorch了,换清华源提示找不到,还是老老实实官网下载吧。
具体下载什么版本,请查阅网址
pytorch以前版本

注意此处需要查阅自己cuda版本,网上教程很多,就不再细说了。
我的是cuda11.3, 查到11.3对应的代码,
cuda 11.3
要是没有翻墙的话,就替换下载源为清华源或者阿里源, 删掉最后的 -c pytorch,否则又是官网下载了。这样安装的是pytorch cpu版本,栓Q。不要上来就调用conda指令安装!
conda虚拟环境中也可以用 pip 安装

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1–extra-index-url https://download.pytorch.org/whl/cu113

这块网址没翻墙的话记得替换为国内源网址,一定记得换
换源教程参考链接GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~

设置清华源加速安装:

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

开始指定版本安装:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html

亲测用pip 指令安装的是pytorch GPU 版本,这一步一坑的实在难绷。要用pip指令安装!!!!
安装完后测试代码,切到python指令模式,输入指令

import torch
torch.cuda.is_available()

测试图片
显示true成功了!

  • 7
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值