torch版本对应关系:
Previous PyTorch Versions | PyTorch
以安装torch==1.7.1,torchvision==0.8.2 ,CUDA 10.1为例
可以尝试直接pip :
pip install torch==1.7.1+cu101
若失败,加入安装源:
pip install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
torchvision同理 :
pip install torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html