使用anaconda安装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/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.1

如果安装失败，与网络延时有关

conda config --set remote_read_timeout_secs 1000.0

python

import torch
x = torch.rand(5, 3)
print(x)