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