tensorrt6.0.1.5 使用报错记录

python import tensorrt 报错如下：

ImportError: libnvinfer.so.6: cannot open shared object file: No such file or directory

可能环境变量没改：

首先 gedit ~/.bashrc，添加如下内容

export LD_LIBRARY_PATH=/home/tensorrt_tar/TensorRT-6.0.1.5/lib:$LD_LIBRARY_PATH

添加的自己解压的路径

然后运行

source ~/.bashrc

才能让其生效

 

编译报错：

NvInfer.h 找不到等

/home/yao/Documents/tensorrtx/yolov5/yololayer.cu(126): error: identifier "PluginTensorDesc" is undefined

/home/yao/Documents/tensorrtx/yolov5/yololayer.cu(126): error: identifier "PluginTensorDesc" is undefined

/home/yao/Documents/tensorrtx/yolov5/yololayer.cu(154): error: identifier "IPluginV2IOExt" is undefined

/home/yao/Documents/tensorrtx/yolov5/yololayer.cu(270): error: identifier "IPluginV2IOExt" is undefined

/home/yao/Documents/tensorrtx/yolov5/yololayer.cu(304): error: identifier "IPluginV2IOExt" is undefined

要在cmakelists.txt 中添加：

include_directories(/home/yao/Downloads/TensorRT-6.0.1.5.Ubuntu-18.04.x86_64-gnu.cuda-10.1.cudnn7.6/TensorRT-6.0.1.5/include)
link_directories(/home/yao/Downloads/TensorRT-6.0.1.5.Ubuntu-18.04.x86_64-gnu.cuda-10.1.cudnn7.6/TensorRT-6.0.1.5/lib)

 

https://github.com/wang-xinyu/tensorrtx/blob/master/tutorials/faq.md#1-fatal-error-nvinferh-no-such-file-or-directory