NVIDIA DALI TensorFlow插件用于CUDA 11.0。Git SHA: 3c97d9b611f242bf455fc79b324b2db53cc6544a
项目描述
TensorFlow插件使DALI与TensorFlow的集成成为可能。
NVIDIA数据加载库(DALI)是一个用于数据加载和预处理的库,旨在加速深度学习应用。它提供了一系列高度优化的构建块,用于加载和处理图像、视频和音频数据。它可以作为流行的深度学习框架内置数据加载器和数据迭代器的便携式替代品。
深度学习应用需要复杂的多阶段数据处理流水线,包括加载、解码、裁剪、缩放和其他许多增强。这些在CPU上执行的数据处理流水线已成为瓶颈,限制了训练和推理的性能和可扩展性。
DALI通过将数据预处理卸载到GPU来解决CPU瓶颈问题。此外,DALI依靠自己的执行引擎,该引擎旨在最大化输入流水线的吞吐量。例如,预取、并行执行和批处理等功能对用户来说都是透明的。
此外,深度学习框架有多个数据预处理实现,导致诸如训练和推理工作流程的可移植性和代码可维护性等挑战。使用DALI实现的数据处理流水线是可移植的,因为它们可以轻松地重定向到TensorFlow、PyTorch、MXNet和PaddlePaddle。
有关更多详细信息,请参阅最新DALI文档。
项目详情
下载文件
下载适合您平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。
源分发
nvidia-dali-tf-plugin-cuda110-1.42.0.tar.gz (384.3 kB 查看哈希值)
关闭
nvidia-dali-tf-plugin-cuda110-1.42.0.tar.gz 的哈希值
算法 | 哈希摘要 | |
---|---|---|
SHA256 | 27fe6f93e28a7298bdeab94139bc0264d0f85863143b4597060429973936de61 |
|
MD5 | b91048e0a19f566eb69b4c6438abf9e7 |
|
BLAKE2b-256 | def0c8d4cdf4d9b000cbf0281bf5567414e529e66580a8ad659ff41fd5461e1d |