pip安装pytorch报错

WARNING: Retrying (Retry(total=4, connect=None

时隔将近一年再去安装pytorch报错,在服务器linux环境下,

看了很多分享,最后试成功举例说:

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

不行,去了以下地址找安装包

https://download.pytorch.org/whl/torch_stable.html

又看了博客还发现,这包还需要和python版本适配,系统装的是python3.9

cu101/torch-1.7.1%2Bcu101-cp39-cp39-linux_x86_64.whl

就是和这里的cp39对应起来,最后找到1.7.1版本的torch,

对应怎么去找torchvision,就是去官网找1.7.1的安装命令,

# CUDA 10.1
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

给出了torchvision的版本,再去下包,同样对应pthon3.9版本

cu101/torchvision-0.8.2%2Bcu101-cp39-cp39-linux_x86_64.whl

最后用命令安装,按分享博客里说的,先装torch后装torchvision,

pip install torch-1.7.1+cu101-cp39-cp39-linux_x86_64.whl

pip install torchvision-0.8.2+cu101-cp39-cp39-linux_x86_64.whl