src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory

src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory

当我按照这条指令安装pycuda时:

pip install pycuda

但是出现错误:

src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory

但是我安装了cuda toolkit.this是nvcc -V

的结果
[root@localhost include]# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 9.2, V9.2.148

这是https://developer.nvidia.com/cuda-downloads

中下载的安装rpm的结果
[root@localhost include]# sudo dnf install cuda
Last metadata expiration check: 0:05:09 ago on Wed 05 Sep 2018 10:08:35 PM EDT.
Package cuda-1:9.2.148.1-2.fc28.x86_64 is already installed, skipping.
Dependencies resolved.
Nothing to do.
Complete!

您可能需要指定 CUDA 的路径:

export C_INCLUDE_PATH=${CUDA_HOME}/include:${C_INCLUDE_PATH}
export LIBRARY_PATH=${CUDA_HOME}/lib64:$LIBRARY_PATH

请确保 echo ${CUDA_HOME} 确实提供了一些合理的输出。

就我而言,我遇到了两个问题:

-lcurand not found

src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory

导出 C_INCLUDE_PATH 对我没有帮助。相反,我需要导出通用版本的 C_INCLUDE_PATH -- CPATH:

export CPATH=$CPATH:/usr/local/cuda/include
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

找出 Cuda 在系统上的安装位置 find / - type d - name cuda 2>/dev/null

使用 or 找到 cuda 的每个位置

export PATH=/usr/local/cuda-VERSION/bin:$PATH

然后

pip install pycuda

这就是我在 Jetson NANO 上解决问题的方法:

sudo pip3 install --global-option=build_ext --global-option="-I/usr/local/cuda-10.0/targets/aarch64-linux/include/" --global-option="-L/usr/local/cuda-10.0/targets/aarch64-linux/lib/" pycuda

这是另一个对我有用的解决方案。此解决方案取自 here.

$ export PATH=/usr/local/cuda/bin:$PATH
$ sudo apt-get install python-dev
$ pip install numpy
$ export CUDA_ROOT=/usr/local/cuda
$ pip install pycuda