@@ -68,9 +68,9 @@ RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
68
68
nodejs \
69
69
pdsh \
70
70
protobuf-compiler \
71
- libsmpeg-dev \
72
- openjdk-8-jdk \
73
- python3-gdal \
71
+ libsmpeg-dev \
72
+ openjdk-8-jdk \
73
+ python3-gdal \
74
74
v4l-utils \
75
75
x264
76
76
@@ -192,23 +192,21 @@ RUN mkdir /tmp/openmpi && \
192
192
make -j $(nproc) all && \
193
193
make install && \
194
194
ldconfig && \
195
- rm -rf /tmp/openmpi*
195
+ rm -rf /tmp/openmpi* && \
196
196
# 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 && \
198
198
echo '#!/bin/bash' > /usr/local/bin/mpirun && \
199
199
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 && \
202
201
# 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
204
203
205
204
RUN python3 -m pip install --no-cache-dir \
206
- mpi4py==3.1.3 \
205
+ mpi4py==3.1.3 \
207
206
nni==2.6.1 \
208
207
mlflow==1.24.0 \
209
208
scikit-nni==0.2.1
210
209
211
-
212
210
# Install Jupyterlab extensions
213
211
RUN jupyter nbextensions_configurator enable && \
214
212
jupyter contrib nbextension install && \
@@ -236,7 +234,7 @@ RUN /usr/bin/python3 -m ipykernel install --display-name "Python 3.8 (NGC 22.03
236
234
237
235
# Backend.AI specifics
238
236
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/
240
238
LABEL ai.backend.kernelspec="1" \
241
239
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
242
240
ai.backend.features="batch query uid-match user-input" \
0 commit comments