参考:https://blog.csdn.net/zong596568821xp/article/details/86077553
一、TensorRT安装
坑:
(1)根据自己cuda的安装方式自行选择下载deb还是tar文件
(2)下载链接中标明了对应的cuda版本以及操作系统
(3)下载后文件名中标明的需要的cuDNN的版本,需要时自行下载
(4)参考博客中的版本只支持到python3.5,python3.6以上版本不支持,即使显示安装成功,在import tensorrt时仍然会报错,若用python3.6版本,建议下载最新版本tensorrt 5.1.5.0,貌似也只支持到python3.7。
附安装步骤:
tar xzvf TensorRT-5.1.5.0.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.5.tar
#解压得到TensorRT-5.0.2.6的文件夹,将里边的lib绝对路径添加到环境变量中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/lthpc/tensorrt_tar/TensorRT-5.1.5.0/lib