Jetson Nano安装pycuda
安装pip:
apt install python3-pip
安装pycuda:
pip3 install pycuda
报错:
In file included from src/cpp/cuda.cpp:4:0:
src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory
#include <cuda.h>
^~~~~~~~
compilation terminated.
error: command 'aarch64-linux-gnu-gcc' failed with exit status 1
----------------------------------------
Failed building wheel for pycuda
实际上CUDA的头文件已经安装在 /usr/local/cuda/targets/aarch64-linux/include/ ，需要让编译make能够找到这些头文件。参考 src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory ，在 /etc/profile 中添加:
export CUDA_HOME=/usr/local/cuda
export CPATH=${CUDA_HOME}/include:${CPATH}
export LIBRARY_PATH=${CUDA_HOME}/lib64:$LIBRARY_PATH
这里使用了 CPATH 变量而不是 C_INCLUDE_PATH ，因为这是一个更为通用的路径输出。我测试了使用 C_INCLUDE_PATH 报错依旧，似乎对于编译c++的 .cpp 应该使用 CPATH 。