@@ -2,16 +2,23 @@ FROM nvcr.io/nvidia/tensorflow:21.03-tf2-py3
2
2
# NVIDIA DIGITS runs on Python 3.8
3
3
4
4
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
+
11
16
ENV DEBIAN_FRONTEND=noninteractive \
12
17
MPLBACKEND=Svg \
13
- PIP_IGNORE_INSTALLED=0 \
18
+ PIP_IGNORE_INSTALLED=0 \
14
19
PYTHONUNBUFFERED=1 \
20
+ CPLUS_INCLUDE_PATH=/usr/include/gdal \
21
+ C_INCLUDE_PATH=/usr/include/gdal \
15
22
LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
16
23
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" \
17
24
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
22
29
# install NLP packages *mecab-ko & khai*
23
30
RUN apt-get update && \
24
31
apt-get install -y \
25
- openjdk-8-jdk \
26
- libasound2-dev \
27
- gfortran \
28
32
automake \
33
+ gfortran \
34
+ libasound2-dev \
35
+ openjdk-8-jdk \
29
36
&& \
30
37
cd /tmp && \
31
38
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 && \
75
82
libgoogle-glog-dev libgflags-dev \
76
83
libgphoto2-dev \
77
84
libeigen3-dev \
78
- libhdf5-dev \
79
- && \
80
- apt-get clean && \
81
- rm -rf /var/lib/apt/lists/
85
+ libhdf5-dev
82
86
83
87
WORKDIR /tmp
84
88
ENV OPENCV_VERSION="4.5.2"
@@ -94,7 +98,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
94
98
-D BUILD_opencv_java=OFF \
95
99
-D WITH_CUDA=ON \
96
100
-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 \
98
102
-D ENABLE_FAST_MATH=1 \
99
103
-D CUDA_FAST_MATH=1 \
100
104
-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
160
164
bash install.sh && \
161
165
rm -rf /tmp/*
162
166
167
+ WORKDIR /tmp
168
+ RUN git clone https://github.com/aristocratos/bashtop.git && \
169
+ cd bashtop && \
170
+ make install
163
171
164
172
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 && \
165
173
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 && \
207
215
208
216
RUN apt autoclean && \
209
217
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 && \
210
219
rm -rf /var/lib/apt/lists/* && \
211
220
rm -rf /root/.cache && \
212
221
rm -rf /tmp/*
213
222
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
+
214
227
# Install ipython kernelspec
215
228
RUN /usr/bin/python3 -m ipykernel install --display-name "Python 3.8 (NGC 21.03 / TensorFlow 2.4) on Backend.AI" && \
216
229
cat /usr/local/share/jupyter/kernels/python3/kernel.json
217
230
218
231
# Backend.AI specifics
219
232
COPY ./service-defs /etc/backend.ai/service-defs
220
- # Copy Backend.Ai multi-node support
221
233
COPY runner-scripts/bootstrap.sh runner-scripts/setup_multinode.py /opt/container/
222
234
LABEL ai.backend.kernelspec="1" \
223
235
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
0 commit comments