1、创建虚拟环境

conda create --name pytorch_gpu python=3.6

虚拟环境名称：pytorch_gpu
python版本：3.6
根据自己的需求做相应的更改
2、激活虚拟环境

conda activate pytorch_gpu

3、更改清华源站

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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

4、安装pytorch
Pytorch官网找到对应的版本

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

以上命令安装的是pytorch1.6 对应cuda10.1的版本
！！！注意！！！ 要把 -c pytorch 去掉，否则还是默认源下载

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1