Ubuntu安装TensorRT6、TensorRT7

1、系统环境确认

• Ubuntu18.04

2、cuda环境确认

• cuda-10.1

# 查看cuda版本
nvcc -V

# 输出
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

•  cudnn7.6

# 查看cudnn版本(cuda10.1的cudnn貌似安装在/usr/include中)
cat /usr/include/cudnn.h |grep CUDNN_MAJOR -A 2
# 其他版本的cuda
 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

# 输出
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

3、下载TensorRT

TensorRT7需要cuda10.2及以上的cuda版本，cuda10.1环境无法安装TensorRT7，故选择TensorRT6进行安装。

下载地址：https://developer.nvidia.com/nvidia-tensorrt-download

如果大家的cuda是安装的runtime版本，那么TensorRT只能选择Tar File Install Packages 下载安装。