NVCC版本与Linux版本不兼容问题

问题描述

使用Ubuntu Server 24.04，python 3.7，cuda 11.1，gcc-10, 安装simple-knn，需要本地编译时nvcc报错 /usr/include/linux/types.h:12:27: error: expected initializer before ‘__s128’。

解决方案

由于内核版本为6.8，cuda 11.1在gcc-10下编译，与linux 5.4版本兼容性较好，因此需要重新下载5.4版linux内核头文件，并且将位置传给nvcc。

下载linux-5.4

cd /usr/local/src
wget https://cdn.kernel.org/pub/linux/kernel/v5.x/linux-5.4.tar.xz
tar -xf linux-5.4.tar.xz
cd linux-5.4
make headers_install INSTALL_HDR_PATH=/usr/local/linux-headers-5.4

设置gcc-10引用路径

export CC=/usr/bin/gcc-10
export CXX=/usr/bin/g++-10
export CFLAGS="-I/usr/local/linux-headers-5.4/include"
export CPPFLAGS="-I/usr/local/linux-headers-5.4/include"

将头文件路径传入到nvcc

在setup.py中extra_compile_args字段，传入给nvcc参数'--system-include=/usr/local/linux-headers-5.4/include'，即设置成
extra_compile_args={"nvcc": ['--system-include=/usr/local/linux-headers-5.4/include'], "cxx": cxx_compiler_flags})