【已解决】PaddlePaddle+TensorRT推理时环境报错

鱼与水人

已于 2023-03-03 21:10:30 修改

阅读量992

点赞数 2

文章标签： paddlepaddle python 深度学习

于 2023-03-01 12:41:41 首次发布

本文链接：https://blog.csdn.net/qq_45607766/article/details/129278903

版权

文章目录

错误提示

PaddlePaddle+TensorRT预测：

The TensorRT that Paddle depends on is not configured correctly.

RuntimeError: (Unavailable) Load tensorrt api getInferLibVersion failed
  [Hint: p_getInferLibVersion should not be null.] (at /paddle/paddle/fluid/platform/dynload/tensorrt.h:114)

运行环境

系统：Ubuntu20.04
python：3.8.0
Cuda：11.3
Cudnn：8.4.0
paddlepaddle-gpu：2.3.2.post112
TensorRT：8.4.1.5
使用PaddleClas有/无安全帽检测官方代码及预训练模型🔗

解决方案

1. 检查安装的PaddlePaddle是否支持TensorRT

下载链接🔗

2. 检查TensorRT的环境变量是否添加

在~/.bashrc中加入了TensorRT的环境变量

sudo gedit ~/.bashrc

# 加入以下三行，只加第一行无效
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/***/TensorRT-8.4.1.5/lib
export TENSORRT_INCLUDE_DIR=/home/***/TensorRT-8.4.1.5
export PATH=/home/***/TensorRT-8.4.1.5/bin:$PATH

# 使改动生效
source ~/.bashrc