可以通过以下步骤来下载适合 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