jetson安装pycuda报错的解决方案

最新推荐文章于 2024-07-30 14:06:46 发布

光元

最新推荐文章于 2024-07-30 14:06:46 发布

阅读量1.5k

点赞数

分类专栏： jetson 文章标签： cuda pip python

本文链接：https://blog.csdn.net/MalusJiang/article/details/115524802

版权

在Jetson设备上使用pip安装pycuda时，可能会遇到找不到cuda.h头文件或numpy版本不匹配的问题。解决方法包括：将CUDA头文件路径添加到LIBRARY_PATH，以及在numpy版本与tensorflow兼容的情况下安装特定版本的pycuda（如2019.1.2）。

摘要由CSDN通过智能技术生成

在使用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: 没有那个文件或目录
在这里插入图片描述