Last Updated:

pytorch安装

yiliang.dong
yiliang.dong python

1、新建pytorch的python环境,避免和其他python环境冲突

conda create -n pytorch

2、打开pytorch官网

https://pytorch.org/get-started/locally/

3、找到适合自己系统的安装命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

上面这个下载速度超级慢

4、添加国内镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/

5、去掉-c pytorch,重新下载才会使用到上面的镜像

conda install pytorch torchvision torchaudio cudatoolkit=11.3

Comments