linux下安装anaconda 和指定版本的gpu版本的pytorch与torchvison(自己记录)

anaconda和miniconda关系

清华镜像:anaconda和miniconda

https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

官方网址

miniconda:    https://docs.conda.io/en/latest/miniconda.html

anaconda:      https://www.anaconda.com/products/distribution

然后直接安装就好

bash Anaconda3-2022.05-Linux-x86_64.sh

安装完配置一下环境变量

    echo 'export PATH="~/anaconda3/bin":$PATH' >> ~/.bashrc
    source ~/.bashrc

‎1.安装pytorch

代码模板:在这里边找想要安装的python版本对应的pytorch版本torchvision版本以及cudatoolkit版本

conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch

          -c pytorch的意思是在pytorch官网安装

          -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/的意思是用清华源安装

          -c https://mirrors.tuna.tsinghua.edu.cn一般是到这里,找所需要的包

放几个源,现用现加

如果用conda 就-c        如果用pip 就 -i

清华源https://mirrors.tuna.tsinghua.edu.cn

豆瓣源https://pypi.douban.com/simple/

中国科技大学https://mirrors.ustc.edu.cn/

阿里云https://developer.aliyun.com/mirror/

推荐:官网安装

官网安装以及历史版本 

PyTorch

官网组成代码

根据相应的cuda版本找安装代码

2.清华源:在这里边找想要安装的python版本对应的pytorch版本torchvision版本以及cudatoolkit版本

安装不成功可以考虑换源

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud     在这里找pytorch或者别的

https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

创建自己的环境:

conda create -n env python==3.9.1

遇到的问题

nvcc -V和nvidia-smi出现cuda版本不一致的情况,在安装pytorch时要根据nvcc-V看到的版本来

https://www.jianshu.com/p/eb5335708f2a

3.根据显卡算力匹配相应的cuda,这个是对照表

而查看算力用nvidia-smi命令

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

4.总结来说就是先查看算力找到对应的cuda版本,再去清华源找相应的pytorch包

尽量用conda安装,因为比较熟悉,虽然pip比较强大

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值