Skip to content

Commit 2ad6053

Browse files
committed
add service-defs
1 parent fd627cd commit 2ad6053

File tree

1 file changed

+28
-16
lines changed

1 file changed

+28
-16
lines changed

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

Lines changed: 28 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,23 @@ FROM nvcr.io/nvidia/tensorflow:21.03-tf2-py3
22
# NVIDIA DIGITS runs on Python 3.8
33

44
RUN apt-get update && \
5-
apt-get install -y libsm6 libxext6 libxrender-dev mercurial libgdal-dev python3-gdal \
6-
libsdl-dev libsdl-image1.2-dev libsdl-mixer1.2-dev libsdl-ttf2.0-dev \
7-
libsmpeg-dev libportmidi-dev libavformat-dev libswscale-dev
8-
9-
ENV CPLUS_INCLUDE_PATH=/usr/include/gdal
10-
ENV C_INCLUDE_PATH=/usr/include/gdal
5+
apt-get install -y --no-install-recommends \
6+
fonts-nanum \
7+
fonts-nanum-coding \
8+
fonts-nanum-extra \
9+
htop \
10+
ncurses-term \
11+
libasound2-dev \
12+
libsm6 libxext6 libxrender-dev mercurial libgdal-dev python3-gdal \
13+
libsdl-dev libsdl-image1.2-dev libsdl-mixer1.2-dev libsdl-ttf2.0-dev \
14+
libsmpeg-dev libportmidi-dev libavformat-dev libswscale-dev
15+
1116
ENV DEBIAN_FRONTEND=noninteractive \
1217
MPLBACKEND=Svg \
13-
PIP_IGNORE_INSTALLED=0 \
18+
PIP_IGNORE_INSTALLED=0 \
1419
PYTHONUNBUFFERED=1 \
20+
CPLUS_INCLUDE_PATH=/usr/include/gdal \
21+
C_INCLUDE_PATH=/usr/include/gdal \
1522
LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
1623
PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/bin/cmake/bin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab" \
1724
mecab_dicdir=/usr/local/lib/mecab/dic/mecab-ko-dic \
@@ -22,10 +29,10 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 2
2229
# install NLP packages *mecab-ko & khai*
2330
RUN apt-get update && \
2431
apt-get install -y \
25-
openjdk-8-jdk \
26-
libasound2-dev \
27-
gfortran \
2832
automake \
33+
gfortran \
34+
libasound2-dev \
35+
openjdk-8-jdk \
2936
&& \
3037
cd /tmp && \
3138
curl -LO https://bitbucket.org/eunjeon/mecab-ko/downloads/mecab-0.996-ko-0.9.2.tar.gz && \
@@ -75,10 +82,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
7582
libgoogle-glog-dev libgflags-dev \
7683
libgphoto2-dev \
7784
libeigen3-dev \
78-
libhdf5-dev \
79-
&& \
80-
apt-get clean && \
81-
rm -rf /var/lib/apt/lists/
85+
libhdf5-dev
8286

8387
WORKDIR /tmp
8488
ENV OPENCV_VERSION="4.5.2"
@@ -94,7 +98,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
9498
-D BUILD_opencv_java=OFF \
9599
-D WITH_CUDA=ON \
96100
-D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr \
97-
-D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.1 \
101+
-D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.2 \
98102
-D ENABLE_FAST_MATH=1 \
99103
-D CUDA_FAST_MATH=1 \
100104
-D WITH_CUBLAS=1 \
@@ -160,6 +164,10 @@ RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v2.13.2/git-l
160164
bash install.sh && \
161165
rm -rf /tmp/*
162166

167+
WORKDIR /tmp
168+
RUN git clone https://github.com/aristocratos/bashtop.git && \
169+
cd bashtop && \
170+
make install
163171

164172
RUN curl -fL https://github.com/cdr/code-server/releases/download/v3.9.0/code-server-3.9.0-linux-amd64.tar.gz | tar -C /usr/local/lib -xz && \
165173
mv /usr/local/lib/code-server-3.9.0-linux-amd64 /usr/local/lib/code-server-3.9.0 && \
@@ -207,17 +215,21 @@ RUN jupyter nbextensions_configurator enable && \
207215

208216
RUN apt autoclean && \
209217
sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
218+
ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color && \
210219
rm -rf /var/lib/apt/lists/* && \
211220
rm -rf /root/.cache && \
212221
rm -rf /tmp/*
213222

223+
# Copy Backend.Ai multi-node support
224+
COPY ./service-defs /etc/backend.ai/service-defs
225+
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
226+
214227
# Install ipython kernelspec
215228
RUN /usr/bin/python3 -m ipykernel install --display-name "Python 3.8 (NGC 21.03 / TensorFlow 2.4) on Backend.AI" && \
216229
cat /usr/local/share/jupyter/kernels/python3/kernel.json
217230

218231
# Backend.AI specifics
219232
COPY ./service-defs /etc/backend.ai/service-defs
220-
# Copy Backend.Ai multi-node support
221233
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
222234
LABEL ai.backend.kernelspec="1" \
223235
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \

0 commit comments

Comments
 (0)