Ubuntu安装TensorRT6、TensorRT7

最新推荐文章于 2024-08-25 11:42:55 发布

ken_asr

最新推荐文章于 2024-08-25 11:42:55 发布

阅读量739

点赞数

分类专栏： linux 机器视觉文章标签： ubuntu nvidia

本文链接：https://blog.csdn.net/zkgoup/article/details/112382662

版权

本文详细介绍了在Ubuntu18.04系统上安装TensorRT6和TensorRT7的步骤，包括系统环境确认、CUDA环境确认、下载TensorRT、tar包安装和deb包安装等过程，特别指出TensorRT7需要CUDA10.2及以上版本。

摘要由CSDN通过智能技术生成

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"