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

这样就安装成功！