清华镜像: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/
推荐:官网安装
官网安装以及历史版本
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比较强大