【torch、torchvision、torchaudio】版本对应关系
·
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
更多推荐
已为社区贡献2条内容
所有评论(0)