可以通过以下步骤来下载适合 CUDA 12.0 版本的 PyTorch：
- 打开命令行/终端。
- 运行以下命令：
pip install torch==1.7.1+cu12 torchvision==0.9.0+cu12 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu12/torch_stable.html
这样就可以安装适合 CUDA 12.0 的 PyTorch 了。如果在安装过程中遇到问题，可以尝试更换源或者联系 PyTorch 官方获得帮助。
本文转载自: https://blog.csdn.net/weixin_42584507/article/details/129509687
版权归原作者 红廉骑士兽 所有， 如有侵权，请联系我们删除。
版权归原作者 红廉骑士兽 所有， 如有侵权，请联系我们删除。