Ubuntu18.04安装TensorRT5.1.5

前言

深度学习中使用到了TesnsorRT，安装过程整理如下

安装TensorRT5.1.5

首先下载对应版本的TensorRT。如TensorRT-5.1.5.0.Ubuntu-18.04.2.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz，压缩包的名称中就包含了cuda以及cudnn的版本，对应即可。
按照下面命令安装即可

$ tar -zxvf TensorRT-5.1.5.0.Ubuntu-18.04.2.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz 
$ cd TensorRT-5.1.5.0/python/
$ sudo apt install python-pip
$ pip install tensorrt-5.1.5.0-cp27-none-linux_x86_64.whl 
$ cd ../graphsurgeon/
$ pip install graphsurgeon-0.4.1-py2.py3-none-any.whl 

之后要记得配置环境

sudo gedit ~/.bashrc

export LD_LIBRARY_PATH=/home/xxxx/TensorRT-5.1.5.0/lib:$LD_LIBRARY_PATH
将上述语句添加到环境变量，xxxx替换为自己的目录名称
，保存退出后使其生效：

source ~/.bashrc