赞
踩
选定版本后,可以采用以下命令安装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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。