ubuntu上tensorrt7.0安装

ubuntu18.04上deepstream5.0开发环境安装,依次安装即可:

1、ubuntu上cuda及显卡驱动安装

2、ubuntu更改屏幕分辨率

3、ubuntu上tensorrt7.0安装

4、ubuntu上dseepstream5.0安装

系统环境:

ubuntu18.04、cuda10.2、driver440、cudnn7.6、tensorRT7.0、python3.6。

首先安装cuda及显卡驱动,之后再依据本文安装tensorrt7.0即可。

1、deb包安装

为了避免安装时各种问题,我们也可以使用deb包的方式安装。下载地址:

https://developer.nvidia.com/nvidia-tensorrt-7x-download

下载文件:nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb

选择自己系统对应的版本,然后使用如下指令安装:

os="ubuntu1804"
tag="cuda10.2-trt7.0.0.11-ga-20191216"
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub
 
sudo apt-get update
sudo apt-get install tensorrt
 
sudo apt-get install python-libnvinfer-dev  # for python2
sudo apt-get install python3-libnvinfer-dev  # for python3
sudo apt-get install uff-converter-tf  # for tensorflow

安装完之后运行下边指令,测试是否成功:

dpkg -l | grep TensorRT

2、tar安装

2.1、tensorRT下载,将tensorRT下载到想要安装的目录

https://developer.nvidia.com/nvidia-tensorrt-download

注:tensorRT官方提供两种安装版本tar和deb安装,但用deb安装多次均未成功,以下只列出了tar安装教程.

2.2、预先安装CUDA10.2 cuDNN7.6 python2/3

2.3、解压tar文件

        解压完会出现TensorRT-7.0.0.11,这个目录下面会包含lib,bin等文件。

tar -xzvf TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn7.6.tar.gz

3、添加环境变量

vim ~/.bashrc 
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:tr7_bin_path # 你的tr7 bin目录
source ~/.bashrc

实例:

export TR_PATH=you tr7 path
export PATH=$PATH:$TR_PATH/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TR_PATH/lib
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TR_PATH/targets/x86_64-linux-gnu/lib

4、安装python tensorRT包

cd tr7/python
pip install tensorrt-7.0.0.11-cp36-none-linux_x86_64.whl

若未安装python,安装python后再安装tr包:

sudo apt-get install python2.7-dev python3.6-dev python-numpy python3-numpy

5、安装UFF包

只有tensorflow需要,pytorch不需要。转到uff目录下:

pip install uff-0.7.5-py2.py3-none-any.whl

一切正常,在第5步就安装完毕了。

传送门:deepstream系列文章分类整理

你可能感兴趣的:(#,deepstream,ubuntu/tx2,ubuntu,cuda,trnsorrt,linux)