@@ -13,8 +13,8 @@ ENV DEBIAN_FRONTEND=noninteractive \
13
13
\
14
14
PATH="/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" \
15
15
mecab_dicdir=/usr/local/lib/mecab/dic/mecab-ko-dic \
16
- MKL_VERSION=2020.1 \
17
- MKL_BUILD=217 \
16
+ MKL_VERSION=2020.3 \
17
+ MKL_BUILD=279 \
18
18
LANG=C.UTF-8
19
19
20
20
# Set default shell to /bin/bash
@@ -96,6 +96,8 @@ RUN curl -sL https://deb.nodesource.com/setup_10.x | bash - && \
96
96
# update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 60 --slave /usr/bin/g++ g++ /usr/bin/g++-7
97
97
98
98
# MKL
99
+ # http://registrationcenter-download.intel.com/akdlm/irc_nas/tec/16903/l_mkl_2020.3.279.tgz
100
+
99
101
WORKDIR /tmp
100
102
RUN mkdir -p /opt/intel/lib && \
101
103
curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION}/download/linux-64/mkl-static-${MKL_VERSION}-intel_${MKL_BUILD}.tar.bz2 | tar xjv && \
@@ -164,10 +166,10 @@ RUN python3 -m pip install pip --no-cache-dir \
164
166
notebook==6.0.3
165
167
166
168
WORKDIR /tmp
167
- COPY ./requirements.20.08 .txt /tmp
168
- RUN python3 -m pip install --no-cache-dir --upgrade -r requirements.20.08 .txt && \
169
+ COPY ./requirements.20.09 .txt /tmp
170
+ RUN python3 -m pip install --no-cache-dir --upgrade -r requirements.20.09 .txt && \
169
171
python3 -m pip uninstall -y tensorboard tensorboard-plugin-wit tensorflow tensorflow-estimator && \
170
- rm -f /tmp/*.whl /tmp/requirements.20.08 .txt
172
+ rm -f /tmp/*.whl /tmp/requirements.20.09 .txt
171
173
172
174
# install NLP packages *mecab-ko & khai*
173
175
RUN apt-get update && \
@@ -291,8 +293,8 @@ RUN git clone --recursive https://github.com/dmlc/xgboost && \
291
293
RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
292
294
git clone --recursive --branch stable --depth 1 https://github.com/microsoft/LightGBM && \
293
295
cd LightGBM && mkdir build && cd build && \
294
- cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. && \
295
- make OPENCL_HEADERS=/usr/local/cuda-11.0/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-11.0/targets/x86_64-linux/lib -j$(nrpoc) && \
296
+ cmake -DUSE_GPU=0 .. && \
297
+ make -j$(nrpoc) && \
296
298
cd /usr/local/src/lightgbm/LightGBM/python-package && \
297
299
python3 setup.py install --precompile
298
300
0 commit comments