使用tengine部署模型时,编译报错:fatal error: NvInfer.h: No such file or directory | #include
很明显,就是缺少头文件,这个头文件是自己配置tengine部署环境中需要安装的TensorRT带的头文件,(有关在Ubuntu系统上配置TensorRT的相关内容,详见我的另一篇博客:Ubuntu18配置CUDA10.2 cudnn8.0.1 TensorRT7.1.3.4)那么就需要在编译部署代码的CMakeList.txt中指定这个头文件的路径,解决方法如下:
先进入到tengine项目根目录:
$ cd Tengine-tengine-lite
编辑CMakeList.txt
sudo vim CMakeList.txt
在CMakeList.txt中适当位置添加如下代码:
# tensorRT
include_directories(/你自己的tensorrt路径/include)
link_directories(/你自己的tensorrt路径/lib)
完事!欢迎关注!!