Nvidia TensorRT系列01-基本介绍

简介: NVIDIA TensorRT 是一个高性能的机器学习推理SDK,支持 TensorFlow、PyTorch 等框架。本文介绍了 TensorRT 的基本概念、安装指南、快速开始、案例和互补软件,如 NVIDIA Triton 推理服务器、DALI 和 TF-TRT。同时,文章还涵盖了 ONNX 支持、版本控制和弃用策略等内容。

Nvidia TensorRT系列01-基本介绍

鉴于Nvidia的在深度学习上已经非常完善,即使有其他的芯片或者是框架,基本思路也都是要借鉴Nvidia的开发的,好比虽然特斯拉网上很多骂声,但是大多数国内的车企还是在发布会上狠狠对比,又好比手机发布会狠狠对比Apple,好的东西总是值得大家去借鉴的,对于小公司而言,使用其他公司的生态框架硬件成本看起来很低,但是由于生态和社区一般,实际上运维起来的成本又很高。所以,希望通过更新这个系列来帮助自己对Nvidia的TensorRT进行一些基础的学习。其中大部分文字借鉴了英文原始的开发文档,也是主要方便对于英文阅读文档有困难的小伙伴能有一些小小的帮助。

NVIDIA TensorRT开发者指南展示了如何使用C++和Python API来实现最常见的深度学习层。它向您展示了如何获取使用深度学习框架构建的现有模型,并使用提供的解析器构建TensorRT引擎。开发者指南还为常见用户任务提供了分步说明,例如创建TensorRT网络定义、调用TensorRT构建器、序列化和反序列化,以及如何使用C++或Python API向引擎提供数据并执行推理;

For previously released TensorRT documentation, refer to the TensorRT Archives.

安装指南: NVIDIA TensorRT Installation Guide

快速开始指南: NVIDIA TensorRT Quick Start Guide i

案例: NVIDIA TensorRT Sample Support Guide

NVIDIA® TensorRT™是一个软件开发工具包(SDK),有助于实现高性能的机器学习推理。它旨在与TensorFlow、PyTorch和MXNet等训练框架相辅相成。它特别关注如何在NVIDIA硬件上快速且高效地运行已经训练好的网络。

对于英伟达的安培架构或者是更高的架构而言。多实例GPU(MIG)是NVIDIA具有NVIDIA Ampere架构或更新架构的GPU的一项功能,它允许用户将单个GPU定向划分为多个较小的GPU。物理分区提供专用的计算和内存切片,具有服务质量保证,并可在GPU的部分上独立执行并行工作负载。对于GPU利用率较低的TensorRT应用程序,MIG可以在对延迟影响较小或没有影响的情况下产生更高的吞吐量。最佳分区方案是特定于应用程序的。

互补软件

NVIDIA Triton™推理服务器是一个更高级别的库,可在CPU和GPU上提供优化的推理功能。它提供了启动和管理多个模型的功能,以及用于提供推理的REST和gRPC端点。

NVIDIA DALI®为预处理图像、音频和视频数据提供了高性能原语。TensorRT推理可以作为DALI管道中的自定义操作符进行集成。可以在此处找到TensorRT推理作为DALI一部分的工作示例。

TensorFlow-TensorRT (TF-TRT) 是将TensorRT直接集成到TensorFlow中的解决方案。它选择TensorFlow图中的子图以通过TensorRT进行加速,而将图的其余部分留给TensorFlow本地执行。结果仍然是一个TensorFlow图,您可以像往常一样执行它。有关TF-TRT的示例,请参阅TensorFlow中的TensorRT示例。

Torch-TensorRT (Torch-TRT) 是一个PyTorch-TensorRT编译器,可将PyTorch模块转换为TensorRT引擎。在内部,首先根据所选的中间表示(IR)将PyTorch模块转换为TorchScript/FX模块。编译器选择PyTorch图的子图以通过TensorRT进行加速,而图的其余部分则由Torch本地执行。结果仍然是一个可以像往常一样执行的PyTorch模块。有关示例,请参阅Torch-TRT示例。

TensorFlow量化工具包提供了用于以降低的精度训练和部署基于Tensorflow 2的Keras模型的实用工具。此工具包专门用于基于操作符名称、类和模式匹配来量化图中的不同层。然后可以将量化后的图转换为ONNX,然后转换为TensorRT引擎。有关示例,请参阅模型库。

PyTorch量化工具包提供了用于以降低的精度训练PyTorch模型的设施,然后可以将其导出以在TensorRT中进行优化。

此外,PyTorch Automatic Sparsity (ASP) 工具提供了用于训练具有结构化稀疏性的模型的设施,然后可以将其导出,并允许TensorRT在NVIDIA Ampere架构GPU上使用更快的稀疏策略。

TensorRT与NVIDIA的分析工具NVIDIA Nsight™ Systems和NVIDIA深度学习分析器(DLProf)集成在一起。

TensorRT的一个受限子集已获准用于NVIDIA DRIVE®产品。某些API仅标记为在NVIDIA DRIVE中使用,不支持一般使用。

ONNX

模型之间的统一语言--ONNX。

TensorRT从框架中导入训练模型的主要方式是通过ONNX交换格式。TensorRT附带了一个ONNX解析器库,以协助导入模型。在可能的情况下,解析器向后兼容至opset 9;ONNX模型Opset版本转换器可以帮助解决不兼容问题。

GitHub上的版本可能比与TensorRT一起提供的版本支持更新的opset。有关支持的opset和操作符的最新信息,请参阅ONNX-TensorRT操作符支持矩阵。对于TensorRT部署,我们建议导出到最新的可用ONNX opset。

TensorRT的ONNX操作符支持列表可以在here.找到。

PyTorch原生支持ONNX导出 ONNX export. 。对于TensorFlow,推荐的方法是使用tf2onnx。

将模型导出到ONNX后,第一步最好是使用Polygraphy运行常量折叠。这通常可以解决ONNX解析器中的TensorRT转换问题,并简化工作流程。有关详细信息,请参阅此示例 this example.。在某些情况下,可能需要进一步修改ONNX模型,例如,用插件替换子图或用其他操作重新实现不支持的操作。为了简化此过程,您可以使用ONNX-GraphSurgeon ONNX-GraphSurgeon.。

版本控制

TensorRT的版本号(MAJOR.MINOR.PATCH)遵循语义版本控制2.0.0标准,适用于其公共API和库ABI。版本号变化如下:

  • 当进行不兼容的API或ABI更改时,MAJOR(主)版本号会变化。
  • 当以向后兼容的方式添加功能时,MINOR(次)版本号会变化。
  • 当进行向后兼容的错误修复时,PATCH(补丁)版本号会变化。

请注意,语义版本控制并不适用于序列化对象。为了重复使用计划文件和计时缓存,主版本、次版本、补丁版本和构建版本(build versions)的版本号必须匹配(NVIDIA DRIVE OS 6.0开发者指南中详细说明了安全运行时的某些例外)。校准缓存通常可以在主版本内重复使用,但兼容性无法保证。

弃用策略

弃用政策用于通知开发者,某些API和工具已不建议使用。从8.0版本开始,TensorRT采用以下弃用政策:

弃用通知将在NVIDIA TensorRT发行说明中公布。

使用C++ API时:

  • API函数将标记为TRT_DEPRECATED_API宏。
  • 枚举将标记为TRT_DEPRECATED_ENUM宏。
  • 所有其他位置将标记为TRT_DEPRECATED宏。
  • 类、函数和对象将有一条声明,记录它们何时被弃用。

使用Python API时,如果使用了已弃用的方法和类,将在运行时发出弃用警告。

TensorRT在弃用后提供12个月的迁移期。

  • 在迁移期间,API和工具将继续工作。
  • 迁移期结束后,将按照语义版本控制的方式删除API和工具。

对于TensorRT 7.x中专门弃用的任何API和工具,12个月的迁移期从TensorRT 8.0 GA发布日期开始计算。

相关实践学习
在云上部署ChatGLM2-6B大模型(GPU版)
ChatGLM2-6B是由智谱AI及清华KEG实验室于2023年6月发布的中英双语对话开源大模型。通过本实验,可以学习如何配置AIGC开发环境,如何部署ChatGLM2-6B大模型。
目录
相关文章
|
机器学习/深度学习 存储 并行计算
一篇就够:高性能推理引擎理论与实践 (TensorRT)
本文分享了关于 NVIDIA 推出的高性能的深度学习推理引擎 TensorRT 的背后理论知识和实践操作指南。
16326 9
一篇就够:高性能推理引擎理论与实践 (TensorRT)
|
12月前
|
并行计算 PyTorch 算法框架/工具
vLLM 架构学习指南
本指南深入解析vLLM高性能推理引擎架构,涵盖核心创新PagedAttention与连续批处理技术,结合代码结构、学习路径与实践建议,系统指导用户从入门到贡献源码的全过程。
6205 5
vLLM 架构学习指南
|
算法 数据可视化 异构计算
SparseGPT:大规模语言模型的一次性精确剪枝——论文解读
SparseGPT提出首个可高效剪枝百亿参数大模型的一次性精确方法,通过稀疏回归与近似求解器实现高稀疏度下仍保持精度,支持半结构化稀疏与量化联合压缩,显著降低推理成本。
1112 3
SparseGPT:大规模语言模型的一次性精确剪枝——论文解读
|
Ubuntu Linux UED
Ubuntu和Debian系统与Hat Linux系列的使用比较
所以,如果你想搞一个家用服务器,或者是个人使用,喜欢新鲜事物,那Ubuntu可能是你的最好选择。如果你需要一个在商业环境中经受住考验的系统,那Fedora或CentOS可能更符合你的口味。不过记住,CentOS现在是CentOS Stream,中途换马了,所以如果你是稳定性的极端追求者,可能得考虑一下这一变化。而Debian,则适合那些对稳定性需求极高,不太追求软件版本新旧的用户。
541 26
|
机器人 API 定位技术
具身智能干货|ROS2理论与实践系列(二):ROS2通信机制核心
机器人是一种高度复杂的系统性实现,一个完整的机器人应用程序可能由若干功能模块组成,每个功能模块可能又包含若干功能点,在不同功能模块、不同功能点之间需要频繁的进行数据交互。比如以导航中的路径规划模块为例: 路径规划时就需要其他功能模块输入数据,并输出数据以被其他模块调用。 输入的数据有地图服务提供的地图数据、定位模块提供的机器人位姿数据、人机交互模块提供的目标点数据......。 输出的路径信息则被运动控制订阅或是回显在人机交互界面上。 那么这些相对独立的功能模块或功能点之间是如何实现数据交互的呢?在此,我们就需要介绍一下ROS2中的通信机制了。
3167 62
|
机器学习/深度学习 编解码 监控
目标检测实战(六): 使用YOLOv8完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
这篇文章详细介绍了如何使用YOLOv8进行目标检测任务,包括环境搭建、数据准备、模型训练、验证测试以及模型转换等完整流程。
30237 59
目标检测实战(六): 使用YOLOv8完成对图像的目标检测任务(从数据准备到训练测试部署的完整流程)
|
设计模式 Java API
05.接口隔离原则介绍
接口隔离原则(ISP)是SOLID原则之一,强调客户端不应依赖于它们不需要的接口。通过将庞大而臃肿的接口拆分为更小、更具体的接口,确保每个接口只包含客户端真正需要的方法,从而提高代码的可维护性和灵活性。本文详细介绍了接口隔离原则的概念、核心思想、实现方式及案例分析,并对比了其与单一职责原则的区别。关键点包括:接口应精简、独立且可扩展,避免强迫实现不必要的方法,减少系统的耦合性。
686 19
|
机器学习/深度学习 存储 人工智能
【AI系统】自定义计算图 IR
本文介绍了模型转换的方法及流程,重点讲解了计算图的自定义方法和优化技术。通过IR(Intermediate Representation)将不同AI框架的模型转换为统一格式,实现跨平台高效部署。计算图由张量和算子构成,支持多种数据类型和内存排布格式,通过算子融合等优化技术提高模型性能。文章还详细说明了如何使用FlatBuffers定义计算图结构,包括张量、算子和网络模型的定义,为自定义神经网络提供了实践指南。
645 3
【AI系统】自定义计算图 IR
|
数据采集 人工智能 算法
视觉语言模型导论:这篇论文能成为你进军VLM的第一步
【6月更文挑战第20天】探索AI如何理解与生成图像和文本,VLM结合图像与文本映射,涉及图像描述、问答等任务。论文由多所名校和机构研究人员共创,介绍VLM历史、类型(对比学习、掩码、生成、预训练)及应用,如图像生成和问答。同时,讨论数据质量、计算资源和模型可解释性的挑战。[阅读更多](https://arxiv.org/pdf/2405.17247)
1067 2
|
消息中间件 测试技术 领域建模
DDD - 一文读懂DDD领域驱动设计
DDD - 一文读懂DDD领域驱动设计
52655 6