Skip to content

Commit dde3a03

Browse files
authored
Update Dockerfile.22.03-tf2-py3
1 parent 7e78f32 commit dde3a03

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

vendor/ngc-tensorflow/Dockerfile.22.03-tf2-py3

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
6868
nodejs \
6969
pdsh \
7070
protobuf-compiler \
71-
libsmpeg-dev \
72-
openjdk-8-jdk \
73-
python3-gdal \
71+
libsmpeg-dev \
72+
openjdk-8-jdk \
73+
python3-gdal \
7474
v4l-utils \
7575
x264
7676

@@ -192,23 +192,21 @@ RUN mkdir /tmp/openmpi && \
192192
make -j $(nproc) all && \
193193
make install && \
194194
ldconfig && \
195-
rm -rf /tmp/openmpi*
195+
rm -rf /tmp/openmpi* && \
196196
# Create a wrapper for OpenMPI to allow running as root by default
197-
RUN mv /usr/local/bin/mpirun /usr/local/bin/mpirun.real && \
197+
mv /usr/local/bin/mpirun /usr/local/bin/mpirun.real && \
198198
echo '#!/bin/bash' > /usr/local/bin/mpirun && \
199199
echo 'mpirun.real --allow-run-as-root "$@"' >> /usr/local/bin/mpirun && \
200-
chmod a+x /usr/local/bin/mpirun
201-
200+
chmod a+x /usr/local/bin/mpirun && \
202201
# Configure OpenMPI to run good defaults:
203-
RUN echo "btl_tcp_if_exclude = lo,docker0" >> /usr/local/etc/openmpi-mca-params.conf
202+
echo "btl_tcp_if_exclude = lo,docker0" >> /usr/local/etc/openmpi-mca-params.conf
204203

205204
RUN python3 -m pip install --no-cache-dir \
206-
mpi4py==3.1.3 \
205+
mpi4py==3.1.3 \
207206
nni==2.6.1 \
208207
mlflow==1.24.0 \
209208
scikit-nni==0.2.1
210209

211-
212210
# Install Jupyterlab extensions
213211
RUN jupyter nbextensions_configurator enable && \
214212
jupyter contrib nbextension install && \
@@ -236,7 +234,7 @@ RUN /usr/bin/python3 -m ipykernel install --display-name "Python 3.8 (NGC 22.03
236234

237235
# Backend.AI specifics
238236
COPY ./service-defs /etc/backend.ai/service-defs
239-
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
237+
COPY ./runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
240238
LABEL ai.backend.kernelspec="1" \
241239
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
242240
ai.backend.features="batch query uid-match user-input" \

0 commit comments

Comments
 (0)