导语: 今天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