国内源安装指定版本Pytorch

一、选择版本

查看本机CUDA版本：

法1：nvcc --version  or  nvcc -V
法2：cat /usr/local/cuda/version.txt

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

 

去Pytorch官网选择合适版本：https://pytorch.org/

如：

pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install torch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1

 

二、国内源网址

清华：https://pypi.tuna.tsinghua.edu.cn/simple

阿里云：https://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

华中理工大学：https://pypi.hustunique.com/

山东理工大学：https://pypi.sdutlinux.org/

豆瓣：https://pypi.douban.com/simple/