1.torch与torchvision版本对应关系
2.torch与torchaudio版本对应关系
选定版本后,可以采用以下命令安装CUDA版本的torch、torchvison、torchaudio
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
ps:cu116代表CUDA11.6 ,可以根据自身情况修改。
参考连接:Installing pre-built binaries — Torchaudio 2.1.0.dev20230529 documentation