use binaries

dockers/base-conda/Dockerfile

@@ -104,6 +104,7 @@ RUN \
     # Install remaining requirements
     pip install -r requirements-extra.txt --upgrade-strategy only-if-needed && \
     pip install -r requirements-test.txt --upgrade-strategy only-if-needed && \
+    pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda${CUDA_VERSION%%.*}0 && \
     rm requirements*
 
 RUN \
@@ -112,8 +113,6 @@ RUN \
     pip install --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex && \
     rm -rf apex
 
-# todo: add DALI
-
 RUN \
     # Show what we have
     pip --version && \

dockers/base-cuda/Dockerfile

@@ -94,6 +94,7 @@ RUN \
 
     # Install all requirements
     pip install -r requirements/devel.txt --upgrade-strategy only-if-needed --use-feature=2020-resolver && \
+    pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda${CUDA_VERSION%%.*}0 && \
     rm -rf requirements*
 
 RUN \
@@ -102,54 +103,6 @@ RUN \
     pip install --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex && \
     rm -rf apex
 
-ENV CMAKE_VERSION=3.16.9
-
-RUN \
-    # need to  install cmake > 3.13 for DALI build
-    apt-get remove cmake -y && \
-    wget https://cmake.org/files/v${CMAKE_VERSION%.*}/cmake-${CMAKE_VERSION}-Linux-x86_64.sh --progress=bar:force:noscroll && \
-    bash cmake-${CMAKE_VERSION}-Linux-x86_64.sh --skip-license && \
-    rm cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
-    cmake --version
-
-RUN \
-    apt-get update -qq && \
-    apt-get install -y \
-        libopencv-dev \
-        libsndfile-dev \
-        libavfilter-dev \
-        libprotobuf-dev \
-        protobuf-compiler \
-    && \
-    # Get DALI source code:
-    git clone --recursive https://github.com/NVIDIA/DALI && \
-    cd DALI && \
-    # Create a directory for CMake-generated Makefiles. This will be the directory, that DALI’s built in.
-    mkdir build && \
-    cd build && \
-    wget https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
-    bash cmake-${CMAKE_VERSION}-Linux-x86_64.sh --skip-license && \
-    rm cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
-    bin/cmake --version && \
-    # Run CMake. For additional options you can pass to CMake, refer to Optional CMake build parameters.
-    bin/cmake -D CMAKE_BUILD_TYPE=Release .. && \
-    # Build. You can use -j option to execute it in several threads
-    make -j"$(nproc)" && \
-    # In order to run DALI using Python API, you need to install Python bindings
-    pip install dali/python && \
-    # claan-up
-    cd ../.. && \
-    rm -rf DALI && \
-    apt-get remove -y \
-        libopencv-dev \
-        libsndfile-dev \
-        libavfilter-dev \
-        libprotobuf-dev \
-        protobuf-compiler \
-    && \
-    apt-get autoremove -y && \
-    apt-get clean && \
-
 RUN \
     # Show what we have
     pip --version && \