Skip to content

Commit c160d9d

Browse files
committed
use binaries
1 parent 57c5f10 commit c160d9d

File tree

2 files changed

+2
-50
lines changed

2 files changed

+2
-50
lines changed

dockers/base-conda/Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ RUN \
104104
# Install remaining requirements
105105
pip install -r requirements-extra.txt --upgrade-strategy only-if-needed && \
106106
pip install -r requirements-test.txt --upgrade-strategy only-if-needed && \
107+
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda${CUDA_VERSION%%.*}0 && \
107108
rm requirements*
108109

109110
RUN \
@@ -112,8 +113,6 @@ RUN \
112113
pip install --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex && \
113114
rm -rf apex
114115

115-
# todo: add DALI
116-
117116
RUN \
118117
# Show what we have
119118
pip --version && \

dockers/base-cuda/Dockerfile

Lines changed: 1 addition & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ RUN \
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

99100
RUN \
@@ -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-
153106
RUN \
154107
# Show what we have
155108
pip --version && \

0 commit comments

Comments
 (0)