CUDA 12.0下运行CuPy代码遇jitify头文件缺失警告的解决方法

运行CuPy的convolve2d_gpu时出现的头文件缺失警告，本质是CuPy的JIT编译器（jitify）找不到CUDA Toolkit的头文件路径，即便重装CUDA和CuPy，也可能因为环境变量未配置或CuPy未正确识别CUDA路径导致问题持续。以下是针对CUDA 12.0、驱动525.60.13环境的解决步骤：

1. 确认CUDA头文件路径 #

CUDA 12.0的头文件默认路径：

• Windows：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\include
• Linux：/usr/local/cuda-12.0/include
• macOS：/usr/local/cuda/include（仅支持CUDA的Mac机型）

检查该路径下是否存在cuda_runtime_api.h、cuda.h、cooperative_groups.h等缺失的头文件，确保CUDA Toolkit安装完整。

2. 配置环境变量 #

Linux/macOS #

终端临时生效：

export C_INCLUDE_PATH=/usr/local/cuda-12.0/include:$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH=/usr/local/cuda-12.0/include:$CPLUS_INCLUDE_PATH
export CUDA_PATH=/usr/local/cuda-12.0

永久生效：将上述命令添加到~/.bashrc（bash）或~/.zshrc（zsh），执行source ~/.bashrc（对应shell）加载配置。

Windows #

打开系统环境变量设置：

• 新建系统变量CUDA_PATH，值为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0
• 编辑Path变量，添加%CUDA_PATH%\include
• 新建C_INCLUDE_PATH和CPLUS_INCLUDE_PATH系统变量，值均为%CUDA_PATH%\include

设置完成后重启终端或IDE，确保环境变量生效。

3. 从源码编译安装CuPy #

若之前用pip安装的是预编译包，可能适配的CUDA版本与你的环境不符。卸载现有包后从源码编译：

pip uninstall cupy-cuda120 cupy -y
pip install cupy --no-binary :all:

编译过程会自动检测本地CUDA路径，生成适配12.0版本的CuPy。

4. 验证CuPy的CUDA路径识别 #

运行以下Python代码，确认CuPy识别的路径与实际CUDA安装路径一致：

import cupy as cp
print("CUDA路径:", cp.cuda.get_cuda_path())
print("头文件路径:", cp.cuda.get_include_path())

若输出路径不符，重新检查环境变量配置。

5. 临时抑制警告（非推荐） #

如果上述方法无法解决且不影响功能，可以临时屏蔽警告：

import warnings
warnings.filterwarnings("ignore", category=FutureWarning)
import cupyx.jit
cupyx.jit.set_jitify_warnings(False)

但建议优先解决路径问题，避免潜在的编译异常。

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