Ubuntu16.04下安装anaconda3+pytorch全过程(离线安装)


一、安装anaconda

到清华源 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 找到对应的安装包下载,本人下载的是Anaconda3-5.3.0-Linux-x86_64.sh
在这里插入图片描述

复制到ubuntu桌面即可,打开终端,使用如下命令:

bash Anaconda3-5.3.0-Linux-x86_64.sh

不断回车,阅读注册信息(不断回车),直到询问是否添加环境变量(如下图),选择yes继续。
在这里插入图片描述
询问是否安装VSCode(如下图),我不需要,选择no继续。

在这里插入图片描述

完成安装,关闭终端,重新打开,使用

conda list

检验是否安装成功。

参考地址:https://blog.csdn.net/u012243626/article/details/82469174

二、创建虚拟环境

将该环境命名为pytorch,使用python37版本。

conda create -n pytorch python=3.7

三、安装Pytorch,Torchvision和Cudatoolkit

同样是去清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/寻找pytorch和torchvision需要版本的安装包。
我选择的是:
Pytorch: pytorch-1.7.1-py3.7_cuda11.0.221_cudnn8.0.5_0.tar.bz2
Torchvision: torchvision-0.8.2-py37_cu110.tar.bz2
然后去conda官网https://anaconda.org/anaconda/cudatoolkit/files下载cudatoolkit。
我选择的是:
Cudatoolkit cudatoolkit-11.0.221-h6bb024c_0.tar.bz2

下载完成后,将三个安装包复制到Ubuntu桌面上即可,打开终端,先激活虚拟环境:

conda activate pytorch

然后在虚拟环境中使用如下命令:

conda install --offline pytorch-1.6.0-py3.6_cuda10.1.243_cudnn7.6.3_0.tar.bz2 
conda install --offline torchvision-0.7.0-py36_cu101.tar.bz2
conda install --offline cudatoolkit-10.1.243-h6bb024c_0.tar.bz2

安装完毕以后,还需要进行一次匹配,需要用到安装命令(如下),但要记得删除“-c pytorch”,否则会优先到官方库上重新下载安装包,而不是优先在本地搜索匹配。

conda install pytorch torchvision cudatoolkit=10.1

完成后,可以试试看能否导入,先启动python:

python

再试着导入torch:

import torch

如果没有错误提醒,成功导入,那就恭喜安装成功了。

参考地址:https://blog.csdn.net/Cbad_code/article/details/109291099


四、注意事项

注意在安装前查清本机配置,如果安装GPU版本的pytorch,则要注意cuda和cudnn的版本。
查询cuda可以用如下命令(注意cat后面是有个空格的):

cat /usr/local/cuda/version.txt

查询cudnn如下:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

最后查得版本,cuda为11.0.207,cudnn为8.1.0
由此可以选择适当的pytorch、torchvision和cudatoolkit版本。且要注意这三者之间版本也要适配。否则可能会出现正常离线安装,但最后一步在线匹配的时候,出现提示,需要将pytorch或torchvision进行退化,这时候就难受了,这也是我踩过的坑,没有办法,只能删环境重来了。

总结

在线安装速度慢、不稳定、易中断,第一次使用在线安装花了差不多一天,因为各种网络问题出错。第二次使用离线安装,从下载anaconda开始,到成功import torch结束,总共用时也就在半小时左右。离线大法好呀!!!!!

  • 5
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值