Skip to main content

NVIDIA DALI TensorFlow plugin for CUDA 12.0. Git SHA: 4562f157a203bd17a1dbc9a0b07f05ba3a41c1fb

Project description

The TensorFlow plugin enables usage of DALI with TensorFlow.

The NVIDIA Data Loading Library (DALI) is a library for data loading and pre-processing to accelerate deep learning applications. It provides a collection of highly optimized building blocks for loading and processing image, video and audio data. It can be used as a portable drop-in replacement for built in data loaders and data iterators in popular deep learning frameworks.

Deep learning applications require complex, multi-stage data processing pipelines that include loading, decoding, cropping, resizing, and many other augmentations. These data processing pipelines, which are currently executed on the CPU, have become a bottleneck, limiting the performance and scalability of training and inference.

DALI addresses the problem of the CPU bottleneck by offloading data preprocessing to the GPU. Additionally, DALI relies on its own execution engine, built to maximize the throughput of the input pipeline. Features such as prefetching, parallel execution, and batch processing are handled transparently for the user.

In addition, the deep learning frameworks have multiple data pre-processing implementations, resulting in challenges such as portability of training and inference workflows, and code maintainability. Data processing pipelines implemented using DALI are portable because they can easily be retargeted to TensorFlow, PyTorch, MXNet and PaddlePaddle.

For more details please check the latest DALI Documentation.

DALI Diagram

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nvidia-dali-tf-plugin-cuda120-1.43.0.tar.gz (384.8 kB view details)

Uploaded Source

File details

Details for the file nvidia-dali-tf-plugin-cuda120-1.43.0.tar.gz.

File metadata

File hashes

Hashes for nvidia-dali-tf-plugin-cuda120-1.43.0.tar.gz
Algorithm Hash digest
SHA256 5f8505b03e7bc0265eca9e869239cec7de256a5eb3f531e856e211eef5c129a2
MD5 f27079fa50d017d5d07214f607b40ce1
BLAKE2b-256 15adfdf358b63320eb20be7c5920658ea1008d9ff80c0bb92909b1a76e6d9dfd

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page