跳转到主要内容

NVIDIA DALI for CUDA 12.0。Git SHA:3c97d9b611f242bf455fc79b324b2db53cc6544a

项目描述

NVIDIA数据加载库(DALI)是一个用于数据加载和预处理的库,用于加速深度学习应用程序。它提供了一系列高度优化的构建块,用于加载和处理图像、视频和音频数据。它可以作为深度学习框架中内置数据加载器和数据迭代器的便携式替代品。

深度学习应用程序需要复杂的多阶段数据处理管道,包括加载、解码、裁剪、缩放和许多其他增强。这些目前由CPU执行的数据处理管道已成为瓶颈,限制了训练和推理的性能和可扩展性。

DALI通过将数据预处理卸载到GPU来解决CPU瓶颈问题。此外,DALI依赖于其自己的执行引擎,旨在最大化输入管道的吞吐量。如预取、并行执行和批处理等特性对用户是透明的。

此外,深度学习框架有多种数据预处理实现,导致训练和推理工作流程的可移植性以及代码可维护性等挑战。使用DALI实现的数据处理管道是可移植的,因为它们可以轻松地重定向到TensorFlow、PyTorch、MXNet和PaddlePaddle。

有关更多详细信息,请参阅最新DALI文档

DALI Diagram

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源代码分发

nvidia_dali_cuda120-1.42.0.tar.gz (1.5 kB 查看哈希值)

上传时间: 源代码

由以下支持