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学习笔记1 - 环境配置_第1张图片TensorRT学习笔记1 - 环境配置_第2张图片
       TensorRT的下载链接。需要登录NVIDIA帐号。我下载的是。不下deb文件的原因是:使用deb安装TensorRT必须满足cuda和cudnn也是使用deb安装的。
       官方安装教程在这里。我使用的是4.3中的方法进行安装,简单易懂。
TensorRT学习笔记1 - 环境配置_第3张图片TensorRT学习笔记1 - 环境配置_第4张图片

你可能感兴趣的:(TensorRT)