9494
9595 # Install all requirements
9696 pip install -r requirements/devel.txt --upgrade-strategy only-if-needed --use-feature=2020-resolver && \
97+ pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda${CUDA_VERSION%%.*}0 && \
9798 rm -rf requirements*
9899
99100RUN \
@@ -102,54 +103,6 @@ RUN \
102103 pip install --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex && \
103104 rm -rf apex
104105
105- ENV CMAKE_VERSION=3.16.9
106-
107- RUN \
108- # need to install cmake > 3.13 for DALI build
109- apt-get remove cmake -y && \
110- wget https://cmake.org/files/v${CMAKE_VERSION%.*}/cmake-${CMAKE_VERSION}-Linux-x86_64.sh --progress=bar:force:noscroll && \
111- bash cmake-${CMAKE_VERSION}-Linux-x86_64.sh --skip-license && \
112- rm cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
113- cmake --version
114-
115- RUN \
116- apt-get update -qq && \
117- apt-get install -y \
118- libopencv-dev \
119- libsndfile-dev \
120- libavfilter-dev \
121- libprotobuf-dev \
122- protobuf-compiler \
123- && \
124- # Get DALI source code:
125- git clone --recursive https://github.com/NVIDIA/DALI && \
126- cd DALI && \
127- # Create a directory for CMake-generated Makefiles. This will be the directory, that DALI’s built in.
128- mkdir build && \
129- cd build && \
130- wget https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
131- bash cmake-${CMAKE_VERSION}-Linux-x86_64.sh --skip-license && \
132- rm cmake-${CMAKE_VERSION}-Linux-x86_64.sh && \
133- bin/cmake --version && \
134- # Run CMake. For additional options you can pass to CMake, refer to Optional CMake build parameters.
135- bin/cmake -D CMAKE_BUILD_TYPE=Release .. && \
136- # Build. You can use -j option to execute it in several threads
137- make -j"$(nproc)" && \
138- # In order to run DALI using Python API, you need to install Python bindings
139- pip install dali/python && \
140- # claan-up
141- cd ../.. && \
142- rm -rf DALI && \
143- apt-get remove -y \
144- libopencv-dev \
145- libsndfile-dev \
146- libavfilter-dev \
147- libprotobuf-dev \
148- protobuf-compiler \
149- && \
150- apt-get autoremove -y && \
151- apt-get clean && \
152-
153106RUN \
154107 # Show what we have
155108 pip --version && \
0 commit comments