TensorRT 3.0安装（tar包安装，找不到cuda.h文件）

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 解压

扫描二维码关注公众号，回复： 2659362 查看本文章

将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就安装成功了