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

 mirrors / pytorch / vision · GitCode

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐