Skip to content

Commit fefcdc0

Browse files
committed
fixed build error caused GPU config
1 parent baa5f32 commit fefcdc0

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

commons/Dockerfile.base.20.09-py36

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@ ENV DEBIAN_FRONTEND=noninteractive \
1313
\
1414
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" \
1515
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 \
1818
LANG=C.UTF-8
1919

2020
# Set default shell to /bin/bash
@@ -96,6 +96,8 @@ RUN curl -sL https://deb.nodesource.com/setup_10.x | bash - && \
9696
# update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 60 --slave /usr/bin/g++ g++ /usr/bin/g++-7
9797

9898
# MKL
99+
# http://registrationcenter-download.intel.com/akdlm/irc_nas/tec/16903/l_mkl_2020.3.279.tgz
100+
99101
WORKDIR /tmp
100102
RUN mkdir -p /opt/intel/lib && \
101103
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 \
164166
notebook==6.0.3
165167

166168
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 && \
169171
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
171173

172174
# install NLP packages *mecab-ko & khai*
173175
RUN apt-get update && \
@@ -291,8 +293,8 @@ RUN git clone --recursive https://github.com/dmlc/xgboost && \
291293
RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
292294
git clone --recursive --branch stable --depth 1 https://github.com/microsoft/LightGBM && \
293295
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) && \
296298
cd /usr/local/src/lightgbm/LightGBM/python-package && \
297299
python3 setup.py install --precompile
298300

0 commit comments

Comments
 (0)