主要原因还是因为没有安装好Tensorrt,最好是安装4.0.1.6,下面以安装5.1.5.0作为代替说明。
Tensorrt 4.0.1.6的资源文件如下
链接: https://pan.baidu.com/s/1NJC0Ol5sLemrxauQI3t9LQ 提取码: fgn1
tar -xzvf TensorRT-5.1.5.0.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/lthpc/tensorrt_tar/TensorRT-5.1.5.0/lib
#安装TensorRT
cd TensorRT-5.0.2.6/python
#if python3
sudo pip3 install tensorrt-5.1.5.0-py2.py3-none-any.whl
#安装UFF
cd TensorRT-5.0.2.6/uff
#if python3
sudo pip3 install uff-0.6.3-py2.py3-none-any.whl
#安装graphsurgeon
cd TensorRT-5.0.2.6/graphsurgeon
#if python3
sudo pip3 install graphsurgeon-0.4.1-py2.py3-none-any.whl
为了避免后边deepstream找不到tensorrt的库,建议把tensorrt的库和头文件添加到系统路径下,这一步很重要
# TensorRT路径下
sudo cp -r ./lib/* /usr/lib
sudo cp -r ./include/* /usr/include