@@ -9,23 +9,22 @@ ENV NCCL=2.8.3
9
9
ENV NCCL_VERSION=2.8.3-1+cuda11.1
10
10
ENV CUDNN_VERSION 8.0.5.43
11
11
ENV MLNX_OFED_VERSION=5.1-2.3.7.1
12
- # Intel MKL
13
12
ENV MKL_VERSION=2020.3
14
13
ENV MKL_BUILD=279
15
- # Python 3.8 is supported by Ubuntu Bionic out of the box
16
- ARG python=3.8
17
- ENV PYTHON_VERSION=${python}
18
14
ENV NUMPY_VERSION 1.19.4
19
15
ENV SCIPY_VERSION 1.5.4
20
16
ENV OPENMPI 4.0.5
17
+ # Python 3.8 is supported by Ubuntu Bionic out of the box
18
+ ARG python=3.8
19
+ ENV PYTHON_VERSION=${python}
21
20
22
21
ENV DEBIAN_FRONTEND=noninteractive \
23
22
MPLBACKEND=Svg \
24
23
PYTHONUNBUFFERED=1 \
25
24
LIBRARY_PATH=/usr/local/cuda/lib64/stubs \
26
25
_CUDA_COMPAT_PATH="/usr/local/cuda/compat" \
27
- LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/include/x86_64-linux-gnu:/opt/intel/compilers_and_libraries_2020.3.279/linux/mkl/lib/intel64" \
28
- PATH="/usr/local/mpi/bin:/usr/local/ nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/bin/cmake/bin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/tensorrt/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab" \
26
+ LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/include/x86_64-linux-gnu:/opt/intel/compilers_and_libraries_2020.3.279/linux/mkl/lib/intel64:/usr/include/x86_64-linux-gnu " \
27
+ 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:/opt/tensorrt/bin:/usr/local/src/lightgbm/LightGBM:/usr/local/bin/mecab" \
29
28
mecab_dicdir=/usr/local/lib/mecab/dic/mecab-ko-dic \
30
29
LANG=C.UTF-8
31
30
@@ -39,7 +38,8 @@ RUN rm -fr /var/lib/apt/lists/* && \
39
38
wget https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB && \
40
39
apt-key add GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB && \
41
40
rm GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB && \
42
- add-apt-repository ppa:deadsnakes/ppa && \
41
+ add-apt-repository ppa:deadsnakes/ppa && \
42
+ add-apt-repository ppa:bashtop-monitor/bashtop && \
43
43
wget https://apt.repos.intel.com/setup/intelproducts.list -O /etc/apt/sources.list.d/intelproducts.list && \
44
44
apt-get update -y && \
45
45
apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
@@ -50,6 +50,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
50
50
dkms \
51
51
git \
52
52
vim \
53
+ htop bashtop \
53
54
yasm \
54
55
wget zip unzip \
55
56
openssh-client openssh-server \
@@ -67,11 +68,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
67
68
libxext6 \
68
69
libxrender-dev \
69
70
libcairo2-dev libgirepository1.0-dev pkg-config gir1.2-gtk-3.0 \
70
- libgdal-dev python3-gdal \
71
- libsdl-dev libsdl-image1.2-dev libsdl-mixer1.2-dev libsdl-ttf2.0-dev \
72
- libsmpeg-dev libportmidi-dev libavformat-dev libswscale-dev \
73
71
libgeos-dev libgeos++-dev \
74
- tesseract-ocr libtesseract-dev \
75
72
pkg-config \
76
73
zlib1g-dev \
77
74
mime-support \
@@ -191,7 +188,6 @@ RUN python3 -m pip install --no-cache-dir \
191
188
tornado==6.1 \
192
189
nvidia-pyindex==1.0.5 \
193
190
matplotlib==3.3.3
194
-
195
191
196
192
WORKDIR /tmp
197
193
RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \
@@ -220,7 +216,7 @@ RUN cd /tmp && \
220
216
echo "library_dirs = /opt/intel/compilers_and_libraries_2020.3.279/linux/mkl/lib/intel64" >> site.cfg && \
221
217
echo "mkl_libs = mkl_rt" >> site.cfg && \
222
218
echo "lapack_libs =" >> site.cfg && \
223
- python3 -m pip install -U --no-cache-dir /tmp/numpy-1.19.4-cp38-cp38-linux_x86_64 .whl && \
219
+ python3 -m pip install -U --no-cache-dir /tmp/numpy-* .whl && \
224
220
python3 setup.py install
225
221
RUN cd /tmp && \
226
222
git clone --recursive https://github.com/bodono/scs-python.git && \
@@ -241,7 +237,6 @@ RUN apt-get update && \
241
237
make -j$(nproc) && \
242
238
make check && \
243
239
make install
244
-
245
240
RUN echo "Install mecab-ko-dic" && \
246
241
cd /tmp && \
247
242
ldconfig && \
@@ -360,7 +355,6 @@ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
360
355
make OPENCL_HEADERS=/usr/local/cuda-11.1/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-11.1/targets/x86_64-linux/lib -j$(nproc) && \
361
356
cd /usr/local/src/lightgbm/LightGBM/python-package && \
362
357
python3 setup.py install --precompile
363
-
364
358
365
359
# Install Open MPI
366
360
RUN mkdir /tmp/openmpi && \
@@ -418,9 +412,6 @@ RUN jupyter nbextensions_configurator enable && \
418
412
jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
419
413
jupyter serverextension enable --py jupyterlab --sys-prefix && \
420
414
jupyter labextension install --no-build @jupyter-widgets/jupyterlab-manager && \
421
- # git clone https://github.com/lambdalisue/jupyter-vim-binding vim_binding && \
422
- # jupyter nbextension enable /home/work/.jupyter/nbextension/vim_binding/vim_binding && \
423
- # jupyter labextension install --no-build @lckr/jupyterlab_variableinspector && \
424
415
jupyter labextension install --no-build @krassowski/jupyterlab-lsp && \
425
416
jupyter serverextension enable --py jupyter_lsp && \
426
417
jupyter labextension install --no-build @jupyterlab/toc && \
0 commit comments