TensorRT installation

Installation TensorRT

TensorRT installed here only to show the tar package. Installation Environment:

  • CUDA 10.0, 7.5 cuDNN

  • ubuntu 18.0.4

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

  • Unzip to a specific directorysudo tar xzvf TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz -C /usr/local

  • Installation TensorRT and related documents

    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
      Will /usr/local/TensorRT-5.1.2.2/targets/x86_64-linux-gnu/libadd to /etc/ld.so.conf, and thensudo ldconfig

If you run

In [1]: import tensorrt as trt

In [2]: 

No problems with the installation was successful. If you are prompted libnvinfer.sonot find most of the third step is not performed.

Published 65 original articles · won praise 26 · Views 100,000 +

Guess you like

Origin blog.csdn.net/bleedingfight/article/details/88896648