柳鲲鹏
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
这样就安装成功!