conda/miniconda 镜像源快速安装pytorch(GPU)_Linux_实录_2022.3.18

conda/miniconda 安装pytorch(GPU)_Linux_实录_2022.3.18

1.添加清华镜像源

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.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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 --set show_channel_urls yes #设置搜索时显示通道地址

2.创建虚拟环境 我在此将其命名为py-gpu

conda create --name py-gpu python=3.8 -y

3.进入创建好的虚拟环境

conda activate py-gpu

4.安装对应版本的pytorch 详见pytorch官网 我的CUDA版本是11.3

conda install pytorch torchvision cudatoolkit=11.3

5.安装完成后验证

python

import torch

torch.cuda.is_available()

退出python编译

exit()

添加pip镜像源 方便后续安装包

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple