Windows 11安装PyCUDA失败：构建pyproject.toml关联Wheel出错

问题重现 #

执行安装命令：

python -m pip install pycuda

触发编译错误：

error: command 'C:\\Program Files (x86)\\Microsoft Visual Studio\\2022\\BuildTools\\VC\\Tools\\MSVC\\14.41.34120\\bin\\HostX86\\x64\\cl.exe' failed with exit code 2
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for pycuda
Failed to build pycuda
ERROR: ERROR: Failed to build installable wheels for some pyproject.toml based projects (pycuda)

环境信息 #

• CUDA版本：12.6.77，nvcc功能正常：

  nvcc: NVIDIA (R) Cuda compiler driver
  Copyright (c) 2005-2024 NVIDIA Corporation
  Built on Thu_Sep_12_02:55:00_Pacific_Daylight_Time_2024
  Cuda compilation tools, release 12.6, V12.6.77
  Build cuda_12.6.r12.6/compiler.34841621_0
  

• CUDA系统路径：

  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6
  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6
  

• 编译工具：Visual Studio 2022 Build Tools（v17.11.5），已集成CUDA开发组件
• Python版本：3.13.0
• Windows版本：11 Pro 23H2 build 22631.4317
• 已安装Python包：

  contourpy       1.3.0
  cycler          0.12.1
  fonttools       4.54.1
  kiwisolver      1.4.7
  matplotlib      3.9.2
  numpy           2.1.2
  packaging       24.1
  pillow          11.0.0
  pip             24.3.1
  pyparsing       3.2.0
  python-dateutil 2.9.0.post0
  setuptools      75.3.0
  six             1.16.0
  wheel           0.44.0
  

解决步骤 #

1. 补全Visual Studio BuildTools组件 #

• 打开Visual Studio Installer，选择「修改」Visual Studio 2022 BuildTools
• 确保勾选以下核心组件：
  • 桌面开发C++（需包含MSVC v143工具集、对应版本Windows SDK）
  • NVIDIA CUDA 12.6开发工具（匹配已安装的CUDA版本）
• 完成安装后重启系统

2. 使用匹配架构的编译终端 #

• 打开x64 Native Tools Command Prompt for VS 2022（而非普通命令提示符），该终端会自动配置64位编译环境变量
• 在该终端内重新执行安装命令：

  python -m pip install pycuda
  

3. 显式指定CUDA路径编译 #

若上述步骤无效，在安装时直接传递CUDA头文件和库路径：

python -m pip install pycuda --global-option=build_ext --global-option="-IC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\include" --global-option="-LC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\lib\x64"

4. 安装预编译wheel文件 #

跳过本地编译，直接使用对应版本的预编译包：

• 下载匹配Python3.13、CUDA12.6的PyCUDA wheel文件（例如pycuda-2024.1+cuda126-cp313-cp313-win_amd64.whl）
• 执行本地安装：

  python -m pip install 本地路径\pycuda-xxxx.whl
  

5. 验证安装结果 #

安装完成后，运行以下代码确认功能正常：

import pycuda.driver as cuda
cuda.init()
print(f"CUDA设备数量：{cuda.Device.count()}")
print(f"当前设备名称：{cuda.Device(0).name()}")

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