Skip to content

Commit eae3cad

Browse files
committed
2 parents 9059713 + 4ac0006 commit eae3cad

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

commons/Dockerfile.base.py38-cuda10.1

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -113,8 +113,8 @@ RUN rm -fr /var/lib/apt/lists/* && \
113113
libtiff-dev \
114114
libzmq3-dev \
115115
pkg-config \
116-
python${PYTHON_VERSION} \
117-
python${PYTHON_VERSION}-dev \
116+
python${PYTHON_VERSION} \
117+
python${PYTHON_VERSION}-dev \
118118
libnccl2=${NCCL_VERSION} \
119119
libnccl-dev=${NCCL_VERSION} \
120120
vim \
@@ -284,7 +284,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
284284
libgstreamer1.0-dev \
285285
libgstreamer-plugins-base1.0-dev \
286286
libgtk-3-dev \
287-
libtbb-dev \
287+
libtbb-dev \
288288
libatlas-base-dev \
289289
libdc1394-22-dev \
290290
libxvidcore-dev \
@@ -358,7 +358,7 @@ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
358358
git clone --recursive --branch stable --depth 1 https://github.com/microsoft/LightGBM && \
359359
cd LightGBM && mkdir build && cd build && \
360360
cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. && \
361-
make OPENCL_HEADERS=/usr/local/cuda-10.2/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-10.2/targets/x86_64-linux/lib -j$(nproc) && \
361+
make OPENCL_HEADERS=/usr/local/cuda-10.1/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-10.1/targets/x86_64-linux/lib -j$(nproc) && \
362362
cd /usr/local/src/lightgbm/LightGBM/python-package && \
363363
python3 setup.py install --precompile
364364

@@ -428,7 +428,6 @@ RUN jupyter nbextensions_configurator enable && \
428428
jupyter nbextension enable execute_time/ExecuteTime && \
429429
jupyter nbextension enable toc2/main && \
430430
jupyter labextension install @jupyterlab/toc && \
431-
# jupyter labextension install @bokeh/jupyter_bokeh && \
432431
jupyter lab build
433432

434433
# Clean up

0 commit comments

Comments
 (0)