如何下载适用于CUDN 12.0版本的PyTorch?

可以通过以下步骤来下载适合 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 官方获得帮助。

物联沃分享整理
物联沃-IOTWORD物联网 » 如何下载适用于CUDN 12.0版本的PyTorch?

发表评论