Windows11下Python3.12.0+CUDA12.1安装PyTorch报错求助

问题根源 #

PyTorch稳定版初期并未提供适配Python 3.12的预编译分发包，这直接导致pip无法找到匹配的torch版本，触发你遇到的报错。

解决办法 #

1. 降级Python到3.11（推荐生产环境） #

Python 3.11是PyTorch稳定版全面支持的版本，适配CUDA 12.1的预编译包完整：

• 卸载当前Python 3.12，安装Python 3.11.x系列版本（建议3.11.6及以上）
• 激活环境后执行PyTorch官网的CUDA 12.1安装命令：

  pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
  

2. 安装PyTorch Nightly开发版（适配Python 3.12） #

若不想降级Python，可尝试Nightly版本（提前支持Python 3.12）：

• 执行以下命令安装适配CUDA 12.1的Nightly包：

  pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
  

• 注意：Nightly版本可能存在未修复的bug，适合开发测试场景。

3. 手动下载对应whl包安装 #

1. 找到PyTorch whl仓库中适配cp312（Python 3.12）和cu121的torch、torchvision、torchaudio包文件
2. 下载后通过pip本地安装，示例命令（替换为你实际下载的文件名）：

pip install torch-2.1.0.dev20231005+cu121-cp312-cp312-win_amd64.whl

额外检查步骤 #

• 先升级pip到最新版本：pip install --upgrade pip
• 确认CUDA 12.1的路径已添加到系统环境变量PATH中
• 若使用虚拟环境，确保已激活目标环境再执行安装命令

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