柳鲲鹏

  • 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

这样就安装成功！