LINUX安裝TensorRT及特别注意事項

柳鲲鵬

• python安裝，參考其他文章。注意，TensorRT4對應的版本号是3.5，高了低了都有可能安裝不了。

這個在TensorRT的python目錄下，可以看到包含有cpXX的whl檔案，就是對用的python版本号。

• 下載下傳
• 解壓

tar xvf TensorRT-4.0.1.6.Ubuntu-16.04.4.x86_64-gnu.cuda-9.2.cudnn7.1.tar.gz       

• 安裝python部分

cd python
sudo pip3 install tensorrt-4.0.1.6-cp35-cp35m-linux_x86_64.whl       

重複：這裡的cp35，表示需要python3.5，高了低了好像都不行。

• 安裝uff部分

cd uff
sudo pip3 install uff-0.4.0-py2.py3-none-any.whl      

這裡對python版本的要求就很寬松，python2/3都可以。

• 修改配置檔案.bashrc

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/home/gh/gh4ai/TensorRT-4.0.1.6/lib:$LD_LIBRARY_PATH

source .bashrc      

• 測試

python3
import tensorrt      

這樣就安裝成功！