搭建Utrnert仓库时PyCUDA安装失败：Failed to build wheels for pycuda

尝试克隆搭建Utrnert GitHub仓库时，pytools、validators等依赖均可正常安装，但pycuda始终安装失败，报错无法构建wheel。以下是针对性解决步骤：

当前环境配置 #

Python 3.7
nvidia-cublas-cu11==11.10.3.66
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cudnn-cu11==9.5.0.50
opencv-contrib-python==4.5.1.48
opencv-python==4.5.1.48
packaging==23.0
Pillow==9.4.0
pkgutil_resolve_name==1.3.10
platformdirs==3.1.1
PyArabic==0.6.15
pycuda==2022.1

安装错误信息 #

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: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects.

解决步骤 #

1. 安装编译依赖 #

pyCUDA构建wheel需要本地编译环境，根据系统安装对应工具：

• Windows：安装Visual Studio，勾选「Desktop development with C++」组件，确保MSVC相关路径加入系统环境变量。
• Linux：安装gcc、g++、make及CUDA开发工具包：

  # Debian/Ubuntu系
  sudo apt-get install gcc g++ make nvidia-cuda-toolkit
  # RHEL/CentOS系
  sudo dnf install gcc gcc-c++ make nvidia-cuda-toolkit
  

• macOS：安装Xcode命令行工具：

  xcode-select --install
  

2. 匹配CUDA与pyCUDA版本兼容性 #

当前环境使用CUDA Runtime 11.7，pyCUDA 2022.1对该版本的支持存在兼容问题，可尝试：

• 安装更适配CUDA 11.x的pyCUDA版本：

  pip install pycuda==2021.1
  

• 若坚持使用2022.1，需确保本地安装的CUDA Toolkit版本与Runtime一致（即CUDA Toolkit 11.7）。

3. 使用预编译wheel跳过编译 #

直接下载对应Python版本、系统架构和CUDA版本的预编译wheel安装，避免编译失败：

# Windows（Python3.7 x64）
pip install pycuda-2022.1-cp37-cp37m-win_amd64.whl
# Linux（Python3.7 x64）
pip install pycuda-2022.1-cp37-cp37m-linux_x86_64.whl

4. 配置CUDA环境变量 #

确保CUDA路径被正确识别：

• Windows：添加系统环境变量CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7，并将%CUDA_PATH%\bin加入PATH。
• Linux：在~/.bashrc或~/.zshrc中添加：

  export PATH=/usr/local/cuda-11.7/bin:$PATH
  export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH
  

  执行source ~/.bashrc使配置生效。

5. 更新pip及构建工具 #

升级pip、setuptools、wheel到最新版本，避免工具版本过低导致的构建问题：

pip install --upgrade pip setuptools wheel

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