添加清华镜像源
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 --set show_channel_urls yes
附:
查看镜像源命令:conda config --show channels
移除镜像源命令:conda config --remove-key channels
安装pytorch
conda install pytorch torchvision cudatoolkit=10.1
验证pytorch
import torch
torch.cuda.is_available()
torch.__version__