@@ -113,8 +113,8 @@ RUN rm -fr /var/lib/apt/lists/* && \
113
113
libtiff-dev \
114
114
libzmq3-dev \
115
115
pkg-config \
116
- python${PYTHON_VERSION} \
117
- python${PYTHON_VERSION}-dev \
116
+ python${PYTHON_VERSION} \
117
+ python${PYTHON_VERSION}-dev \
118
118
libnccl2=${NCCL_VERSION} \
119
119
libnccl-dev=${NCCL_VERSION} \
120
120
vim \
@@ -284,7 +284,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
284
284
libgstreamer1.0-dev \
285
285
libgstreamer-plugins-base1.0-dev \
286
286
libgtk-3-dev \
287
- libtbb-dev \
287
+ libtbb-dev \
288
288
libatlas-base-dev \
289
289
libdc1394-22-dev \
290
290
libxvidcore-dev \
@@ -358,7 +358,7 @@ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
358
358
git clone --recursive --branch stable --depth 1 https://github.com/microsoft/LightGBM && \
359
359
cd LightGBM && mkdir build && cd build && \
360
360
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) && \
362
362
cd /usr/local/src/lightgbm/LightGBM/python-package && \
363
363
python3 setup.py install --precompile
364
364
@@ -428,7 +428,6 @@ RUN jupyter nbextensions_configurator enable && \
428
428
jupyter nbextension enable execute_time/ExecuteTime && \
429
429
jupyter nbextension enable toc2/main && \
430
430
jupyter labextension install @jupyterlab/toc && \
431
- # jupyter labextension install @bokeh/jupyter_bokeh && \
432
431
jupyter lab build
433
432
434
433
# Clean up
0 commit comments