一、下载pytorch离线.whl包
pytorch 离线.whl包官方下载地址:
https://download.pytorch.org/whl/torch_stable.html
以python 3.6 为例,找到相应版本的torch
、torchaudio
、torchvision
:
torch-1.10.2+cu113-cp36-cp36m-linux_x86_64.whl
torchaudio-0.10.2+cu113-cp36-cp36m-linux_x86_64.whl
torchvision-0.11.1+cu113-cp36-cp36m-linux_x86_64.whl
下载完成后,上传至Linux服务器上。
二、安装.whl包
pip install torch-1.10.2+cu113-cp36-cp36m-linux_x86_64.whl
pip install torchaudio-0.10.2+cu113-cp36-cp36m-linux_x86_64.whl
pip install torchvision-0.11.1+cu113-cp36-cp36m-linux_x86_64.whl --no-deps
注意: 直接安装 pip install torchvision-0.11.1+cu113-cp36-cp36m-linux_x86_64.whl
会安装相关依赖(重新安装 torch):
使用--no-deps
选项表示仅安装torchvision。