TensorRT (一） Ubuntu 安裝配置

環境：

• Ubuntu 18.04
• CUDA 10
• CUDNN 7.6.3
• TensorRT 7

  

一、 下載tar包

TensorRT官方：地址

 

二、 解壓tar包

$ tar xzvf TensorRT-7.x.x.x.<os>.<arch>-gnu.cuda-x.x.cudnn7.x.tar.gz
-其中
    6.x.x.x is your TensorRT version
    <os> is:
        Ubuntu-14.04
        Ubuntu-16.04
        Ubuntu-18.04
        CentOS-7.6
    <arch> is x86_64 or ppc64le
    cuda-x.x is CUDA version 9.0, 10.0, or 10.1
    cudnn7.x is cuDNN version 7.6

 

三、 添加環境變量

vi ~/.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/catt/tensorRT_.70/TensorRT-7.0.0.11/lib

source ~/.bashrc

 

四、 安裝TensorRT的python接口

cd TensorRT-7.x.x.x/python
pip install tensorrt-7.x.x.x-cp3x-none-linux_x86_64.whl

 

五、安裝UFF（Tensorflow所使用的）

cd TensorRT-7.x.x.x/uff
pip install uff-0.7.5-py2.py3-none-any.whl

 

六、 安裝graphsurgeon

cd TensorRT-7.x.x.x/graphsurgeon
pip install graphsurgeon-0.4.1-py2.py3-none-any.whl