1、cuda11.1

https://pytorch.org/

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

2、cuda 11.3

官网：https://pytorch.org/
历史版本：https://pytorch.org/get-started/previous-versions/

升级cuda到11.3
https://www.iotword.com/1971.html

命令行安装依赖
pip install torch==1.10.1+cu113 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113


测试：
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)