TensorRT学习笔记1 - 环境配置

版本选择

       ubuntu 16.04LTS
       tensorrt 5.1.5
       cuda 9.0.176
       cudnn 7.5.0
       python 3.5.2
       gcc 5.4.0

安装过程

       首先，cuda、cudnn、python和gcc的版本一定要和选择的TensorRT版本对应，可在此链接查看。下图为TensorRT5.1.5所需的版本。

       TensorRT的下载链接。需要登录NVIDIA帐号。我下载的是<TensorRT 5.1.5.0 GA for Ubuntu 16.04 and CUDA 9.0 tar package>。不下deb文件的原因是：使用deb安装TensorRT必须满足cuda和cudnn也是使用deb安装的。
       官方安装教程在这里。我使用的是4.3中的方法进行安装，简单易懂。