pytorch安装新坑

问题背景

入坑深度学习有小半年了(菜狗一个大佬不要笑哈),pytorch成为了我不可分割的一部分,单是安装pytorch便不下20次了,最近新来的一个服务器,第一步肯定是按部就班的配环境,然后就可以跑一些实验了,没想到pytorch又出来坑人了。

问题与分析

问题其实很简单,服务器中cuda什么的已经装好,只需要按照cuda版本到pytoch官网找下载命令即可,输入命令:

nvidia-smi

服务器cuda版本
按照cuda版本找到目前最新的安装命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

开始出现第一个问题,这是走了2min的下载进度:
第一个问题:下载速度满
但是我们不慌,换源呗,在网上找了清华源、中科大源,将镜像的路径放在/home/自己用户名/.condarc下面,然后根据网上大哥指点去掉下载命令中的 -c pytorch(这样才能使命令找国内源):

conda install pytorch torchvision torchaudio cudatoolkit=11.3

按常理来说,到这里也没啥问题就结束了,结果出现了第二个问题,报错的原因是在几个国内源找不到对应版本的torchaudio:
第二个问题:找不到对应包
然后我开始尝试用pip下载对应版本,发现也找不到,无可奈何只能继续在conda下尝试,既然这个包在国内这几个源找不到,那就先单独用原命令下载:

conda install torchaudio==0.10.1 -c pytorch

结果居然还是报错了,但这个报错同时给我了关键的信息:
第三个问题
其实挺惊讶的,python现在都已经3.10了,只能怪自己孤陋寡闻了我,老老实实降版本,然后重新运行刚才的命令:

conda install torchaudio==0.10.1 -c pytorch

到这里感觉基本可以解决问题了,然后我又看到了
第一个问题:下载速度满
好像确实还没解决最开始的问题,兜兜转转回到原点,国内源又找不到对应的包,官网的源又太慢还经常断线,正当我陷入瓶颈时,同学跟我说他在原来服务器也是用同样的方式换源并没有问题呀!我突然想到:不会人家的国内源用的跟我不一样?!
不找不知道,一找吓一跳:
北外镜像源
果然还是见识少的惹的祸啊,感谢北外镜像源啊!直接在自己的.condarc添加上这两句:

-https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
-https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/

接着就是回到刚才那句命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3

安装完成后例行检查一下pytorch是否安装成功,进入python环境:

$ >> import torch
$ >> torch.cuda.is_available()
True

完结撒花!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Tensor…Flow

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值