怎么下载适合cudn12.0版本的pytorvh

可以通过以下步骤来下载适合 CUDA 12.0 版本的 PyTorch:

  1. 打开命令行/终端。
  2. 运行以下命令:
    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 官方获得帮助。

猜你喜欢

转载自blog.csdn.net/weixin_42584507/article/details/129509687