在使用pip安装pycuda的时候可能会遇见好几种报错
1.找不到cuda.h头文件
首先是最常见的
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
遇见这种报错是因为找不到cuda.h文件,然而这个文件在jetson是可以找到的,默认放在了cuda的文件夹里
/usr/local/cuda-10.2/targets/aarch64-linux/include
所以只需要手动将路径添加到LIBRARY_PATH即可,指令如下
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.2/targets/aarch64-linux/libexport
export CPATH=$CPATH:/usr/local/cuda-10.2/targets/aarch64-linux/include
2.numpy版本不匹配
第二种可能遇见的报错是
numpy/core/src/multiarray/numpyos.c:18:10: fatal error: xlocale.h: 没有那个文件或目录

遇到这个报错是因为numpy的版本不匹配,使用pip安装pycuda的依赖库numpy版本默认为较新的版本。<

在Jetson设备上使用pip安装pycuda时,可能会遇到找不到cuda.h头文件或numpy版本不匹配的问题。解决方法包括:将CUDA头文件路径添加到LIBRARY_PATH,以及在numpy版本与tensorflow兼容的情况下安装特定版本的pycuda(如2019.1.2)。
最低0.47元/天 解锁文章
1654

被折叠的 条评论
为什么被折叠?



