[问题已处理]-dockerfile构建安装pycuda的时候报错 fatal error- cuda.h- No such file or directory

导语： 今天dockerfile构建安装pycuda的时候报错 fatal error: cuda.h: No such file or directory

因为dockerfile中已经安装了cuda 执行命令，先看看是否存在cuda.h

find /usr/local/cuda-11.2 -name cuda.h

创建测试的文件has_cuda.c 内容如下

#include<cuda.h>

int main ()
{
    int deviceCount;
    cudaError_t e = cudaGetDeviceCount(&deviceCount);
    return e == cudaSuccess ? deviceCount : -1;
}

执行命令测试

gcc has_cuda.c 

发现因为我没有设置环境变量，设置如下环境变量后就可以构建了。

export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.2/targets/x86_64-linux/lib 
export CPATH=\$CPATH:/usr/local/cuda-11.2/targets/x86_64-linux/include 

也有可能需要安装nvidia-cuda-toolkit，或者copy对应的libEGL_nvidia.so文件 如libEGL_nvidia.so.470.63.01

apt-get install nvidia-cuda-toolkit -y