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