Ubuntu下pip安装pycuda报错：无法构建pycuda wheels

出现/usr/bin/x86_64-linux-gnu-gcc' failed with exit code 1错误，本质是pycuda源码编译时缺少必要依赖，以下是可行的解决步骤：

1. 安装基础编译工具与Python开发依赖 #

编译pycuda需要完整的C语言编译链和Python头文件，执行以下命令安装：

sudo apt update && sudo apt install build-essential python3-dev

2. 安装CUDA相关依赖 #

pycuda依赖CUDA库进行编译，安装Ubuntu官方维护的CUDA工具包：

sudo apt install nvidia-cuda-toolkit

3. 配置环境变量（可选，若仍编译失败） #

如果安装依赖后还是报错，手动指定CUDA头文件和库的路径：

export C_INCLUDE_PATH=/usr/include/cuda:$C_INCLUDE_PATH
export LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LIBRARY_PATH

4. 重新安装pycuda #

完成上述步骤后，执行安装命令：

pip install pycuda

替代方案：直接安装预编译包 #

如果不想源码编译，可直接用apt安装Ubuntu打包好的pycuda版本：

sudo apt install python3-pycuda

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