paddle报ValueError: (InvalidArgument) Pass tensorrt_subgraph_pass has not been registered

Posted 2022-07-12 修炼之路

错误提示

在使用paddleOCR做TensorRT优化的时候报了下面的错误

ValueError: (InvalidArgument) Pass tensorrt_subgraph_pass has not been registered

运行环境

• 系统：ubuntu16.04
• python：3.8.10
• TensorRT：7.2.2.3
• cuda：11.1
• cudnn：8.1
• paddlepaddle：2.3.0.post111

解决办法

1. TensorRT是否安装成功

#在bashrc中添加TensorRT的环境变量
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/***/TensorRT-7.2.2.3/lib
export TENSORRT_INCLUDE_DIR=/home/***//TensorRT-7.2.2.3
export PATH=/home/***/TensorRT-7.2.2.3/bin:$PATH
#使环境变量生效
source ~/.bashrc

#安装python版本的TensorRT
cd /home/***/TensorRT-7.2.2.3/python
#选择python版本一致的TensorRT进行安装
pip install tensorrt-7.2.2.3-cp38-none-linux_x86_64.whl
#安装其他的依赖
cd /home/***/TensorRT-7.2.2.3/uff
pip install uff-0.6.9-py2.py3-none-any.whl
cd /home/***/TensorRT-7.2.2.3/graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

#查看TensorRT是否安装成功
>>> import tensorrt
>>> tensorrt.__version__
'7.2.2.3'

2. 安装的paddlepaddle是否支持TensorRT

paddlepaddle-tensorRT:下载链接

#选择与自己服务器环境相匹配的版本
pip install *.whl