1.进入Pytorch官网查看版本对应关系
Previous PyTorch Versions | PyTorch(官网)
2.pip安装即可使用
pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
pip install torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
pip install torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
(cu111对应CUDA版本为CUDA11.1)其他版本同理。
安装出现torchvision没有找到对应版本,安装报错提示里面对应版本即可。
import torch
print(torch.__version__)
print(torch.cuda_version)
print(torch.cuda.is_available())
*****显示结果*****
1.9.1+cu111
11.1
True