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

conda install pytorch torchvision cudatoolkit=10.0

验证GPU可用:

>>import torch

>>print(torch.cuda.is_available())

True

猜你喜欢

转载自www.cnblogs.com/Concerning/p/10813049.html