ubuntu18.04上deepstream5.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步就安装完毕了。