Installation¶

Installing prebuilt DALI packages¶

Linux x64
NVIDIA Driver supporting CUDA 9.0 or later (i.e., 384.xx or later driver releases)
One or more of the following Deep Learning frameworks:
- MXNet 1.3 beta mxnet-cu90==1.3.0b20180612 or later
- pyTorch 0.4
- TensorFlow 1.7 or later

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali

Linux x64
NVIDIA CUDA 9.0 (CUDA 8.0 compatibility is provided unofficially)
nvJPEG library (This can be unofficially disabled; see below)
protobuf version 2 or later (version 3 or later is required for TensorFlow TFRecord file format support)
CMake 3.5 or later
libjpeg-turbo 1.5.x or later (This can be unofficially disabled; see below)
OpenCV 3 or later (OpenCV 2.x compatibility is provided unofficially)
(Optional) liblmdb 0.9.x or later
One or more of the following Deep Learning frameworks:
- MXNet 1.3 beta mxnet-cu90==1.3.0b20180612 or later
- pyTorch 0.4
- TensorFlow 1.7 or later

Note

TensorFlow installation is required to build the TensorFlow plugin for DALI

Note

Items marked “unofficial” are community contributions that are believed to work but not officially tested or maintained by NVIDIA.

git clone --recursive https://github.com/NVIDIA/dali
cd dali

mkdir build
cd build

To build DALI without LMDB support:

cmake ..
make -j"$(nproc)" install

To build DALI with LMDB support:

cmake -DBUILD_LMDB=ON ..
make -j"$(nproc)" install

Optional CMake build parameters:

pip install dali/python