TensorRT 3.0安装
最近leader分配了deepstream的研究任务
deepstream的原理并不是太难,主要深度学习的加速框架,它的突出特性就是能够在深度学习的整个应用流程中提供加速服务,例如视频的处理流程中,从编解码到视频分析再到视频显示的整个流程中,都可以利用deepstream提供的插件来对相应流程进行加速。
要想理解deepstream,就要先配置好环境,deepstream环境依赖如下:
其他的都好办,唯有TensorRT3.0安装起来比较麻烦。虽然有安装说明,但却遇到以下问题
下载 TensorRT-3.0.4.Ubuntu-16.04.3.x86_64.cuda-9.0.cudnn7.0.tar.gz
tar -zxvf TensorRT-3.0.4.Ubuntu-16.04.3.x86_64.cuda-9.0.cudnn7.0.tar.gz 解压
将tensorrt的lib增加到环境变量中
export LD_LIBRARY_PATH=/opt/TensorRT-3.0.4/lib/:${LD_LIBRARY_PATH}
cd TensorRT-3.x.x/python
pip install tensorrt-3.0.4-cp35-cp35m-linux_x86_64.whl
但是在安装pycuda时会遇到cuda.h文件找不到的问题
究其原因是gcc编译时没有查找到cuda.h文件,因此只要将cuda.h文件路径添加到gcc的头文件搜索路劲中即可:
export C_INCLUDE_PATH=/usr/local/cuda-9.0/include/:${C_INCLUDE_PATH}
export CPLUS_INCLUDE_PATH=/usr/local/cuda-9.0/include/:${CPLUS_INCLUDE_PATH}
然后再执行pip install tensorrt-3.0.4-cp35-cp35m-linux_x86_64.whl就可以很快安装好了,
之后还要安装uff包
cd TensorRT-3.x.x/uff
pip install uff-0.2.0-py2.py3-none-any.whl
这样TensorRT就安装成功了