对于安装pycuda安装 cuda.h文件找不到的问题,我之前安装tensorrt遇到过,不过最近我帮同事安装时也遇到了这个问题,但原来的方法无效,只好仔细分析问题,目前分析出来问题原因有三点。
(1)cuda.h这个文件的确找不到,只要添加cuda.h的搜索路径即可。
https://blog.youkuaiyun.com/u011337602/article/details/81238164
(2)未添加CUDA_INC_DIR到/etc/profile
在/etc/profile最后两行添加 export CUDA_INC_DIR=$CUDA_INC_DIR:/usr/local/cuda-9.0/include
保存并执行source /etc/profile
(3)安装pycuda使用的是sudo pip3 install pycuda
这种情况下,执行的root权限,但由于pycuda安装时会使用nvcc,而nvcc是不具有root权限的,所以会导致安装失败。
因此需要去掉sudo,只在用户权限下执行pip3 install pycuda,即可安装成功。