Ubuntu22.04 Anaconda环境安装pyCUDA失败求助

问题核心 #

执行pip install pycuda时出现两个关键错误：

1. 提示未运行configure.py，将使用默认配置（大概率导致编译失败）
2. gcc报错：fatal error: cannot execute ‘cc1plus’: execvp: No existe el archivo o el directorio（缺少C++编译核心组件）

已知环境：

• CUDA 11.8已安装，nvcc --version可正常输出版本信息
• PATH已配置/usr/local/cuda-11.8/bin路径
• 使用Anaconda的openmmlab虚拟环境

解决步骤 #

1. 修复cc1plus缺失问题 #

cc1plus是GNU C编译器（g）的核心组件，执行以下命令安装完整编译工具链：

sudo apt update
sudo apt install build-essential g++ python3-dev

2. 正确配置并安装pyCUDA #

先激活目标Anaconda虚拟环境：

conda activate openmmlab

方法一：环境变量指定CUDA路径快速安装

直接在pip安装时传入CUDA根目录环境变量，自动完成配置：

CUDA_ROOT=/usr/local/cuda-11.8 pip install pycuda

方法二：源码编译安装（配置更可控）

下载pyCUDA源码包并解压，进入源码目录后执行以下命令：

python configure.py --cuda-root=/usr/local/cuda-11.8
pip install .

验证安装结果 #

安装完成后，在Python环境中执行以下代码验证：

import pycuda.driver as cuda
cuda.init()
print(f"CUDA设备数量：{cuda.Device.count()}")

若正常输出设备数量，则说明pyCUDA安装成功。

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