TensorRT安装

安装TensorRT

TensorRT安装这里只展示tar包。安装环境：

• CUDA 10.0,cuDNN 7.5

• ubuntu 18.0.4

• 下载TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz

• 解压到指定目录sudo tar xzvf TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz -C /usr/local

• 安装TensorRT和相关文件

  1. pip install /usr/local/TensorRT-5.1.2.2/python/tensorrt-5.1.2.2-cp36-none-linux_x86_64.whl
  2. pip install /usr/local/TensorRT-5.1.2.2/uff/uff-0.6.3-py2.py3-none-any.whl
  3. pip install /usr/local/TensorRT-5.1.2.2/graphsurgeon/graphsurgeon-0.4.0-py2.py3-none-any.whl
     将/usr/local/TensorRT-5.1.2.2/targets/x86_64-linux-gnu/lib添加到/etc/ld.so.conf，然后sudo ldconfig

如果运行

In [1]: import tensorrt as trt

In [2]: 

没有出现问题则安装成功。如果提示libnvinfer.so没有找到多半是第三步没有执行。