Win10 64位CUDA12.0环境下安装PyTorch的可行方案咨询

无需使用NGC容器或从源码编译，以下两种方式是最简便的解决方案：

方式一：使用pip安装 #

• 打开命令提示符（CMD）或PowerShell，确保已配置好Python环境（推荐3.8-3.11版本）
• 执行以下命令安装支持CUDA12.0的PyTorch 2.x版本：
  pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

注：PyTorch的cu121包兼容CUDA12.0环境，无需升级本地CUDA版本，官方已做向下兼容处理

方式二：使用conda安装 #

• 打开Anaconda Prompt或Miniconda Prompt
• 执行以下命令：
  conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

注：pytorch-cuda=12.1的包可直接在CUDA12.0环境下运行，nvidia频道会自动处理依赖兼容

验证安装成功 #

在Python环境中执行以下代码：

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

若输出True及对应CUDA版本（如12.1），则说明安装成功且CUDA加速功能正常。

内容的提问来源于stack exchange，提问作者Omer B