@@ -305,7 +305,7 @@ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
305
305
git clone --recursive --branch stable --depth 1 https://github.com/microsoft/LightGBM && \
306
306
cd LightGBM && mkdir build && cd build && \
307
307
cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. && \
308
- make OPENCL_HEADERS=/usr/local/cuda-11.2 /targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-11.2 /targets/x86_64-linux/lib -j$(nproc) && \
308
+ make OPENCL_HEADERS=/usr/local/cuda-11.7 /targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-11.7 /targets/x86_64-linux/lib -j$(nproc) && \
309
309
cd /usr/local/src/lightgbm/LightGBM/python-package && \
310
310
python3 setup.py install --precompile
311
311
@@ -447,13 +447,37 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
447
447
ldconfig
448
448
449
449
# Install Jupyterlab extensions
450
- RUN jupyter lab build --dev-build=False --minimize=False && \
451
- python3 -m pip install --no-cache-dir \
450
+ RUN python3 -m pip install --no-cache-dir \
452
451
jupyter_nbextensions_configurator \
453
452
jupyter_contrib_nbextensions \
454
453
jupyter_lsp \
455
- torchvision \
454
+ jupyterlab \
455
+ jupyterlab-code-formatter==1.4.10 \
456
+ jupyterlab-hdf==0.5.1 \
457
+ jupyterlab-launcher==0.13.1 \
458
+ jupyterlab-lsp==3.6.0 \
459
+ jupyter-contrib-core==0.3.3 \
460
+ jupyter-contrib-nbextensions==0.5.1 \
461
+ jupyter-dash==0.4.0 \
462
+ jupyter-highlight-selected-word==0.2.0 \
463
+ jupyter-js-widgets-nbextension==0.0.2.dev0 \
464
+ jupyter-kite==1.2.0 \
465
+ jupyter-latex-envs==1.4.6 \
466
+ jupyter-lsp==1.2.0 \
467
+ jupyter-nbextensions-configurator==0.4.1 \
468
+ jupyter-packaging==0.9.2 \
469
+ jupyter-server-mathjax \
470
+ jupyter-server-proxy==1.6.0 \
471
+ jupyter-server \
472
+ jupyter-telemetry==0.1.0 \
473
+ jupyter==1.0.0 \
474
+ jupyter_bokeh==2.0.4 \
475
+ jupyterhub==2.2.2 \
476
+ jupyterhub-jwtauthenticator==0.1 \
477
+ jupyterhub-kubespawner==0.16.1 \
478
+ jupyterhubutils==0.25.0 \
456
479
ipywidgets && \
480
+ jupyter lab build --dev-build=False --minimize=False && \
457
481
jupyter nbextensions_configurator enable && \
458
482
jupyter contrib nbextension install && \
459
483
jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
@@ -467,6 +491,14 @@ RUN jupyter lab build --dev-build=False --minimize=False && \
467
491
jupyter labextension install @jupyterlab/toc && \
468
492
jupyter lab build
469
493
494
+ RUN curl -fL https://github.com/cdr/code-server/releases/download/v4.9.1/code-server-4.9.1-linux-amd64.tar.gz \
495
+ | tar -C /usr/local/lib -xz && \
496
+ mv /usr/local/lib/code-server-4.9.1-linux-amd64 /usr/local/lib/code-server-4.9.1 && \
497
+ ln -s /usr/local/lib/code-server-4.9.1/bin/code-server /usr/local/bin/code-server && \
498
+ code-server \
499
+ --install-extension ms-python.python \
500
+ --install-extension ms-ceintl.vscode-language-pack-ko
501
+
470
502
# Backend.AI specifics
471
503
# Copy Backend.Ai multi-node support
472
504
COPY ./policy.yml /etc/backend.ai/jail/policy.yml
@@ -487,7 +519,7 @@ LABEL ai.backend.kernelspec="1" \
487
519
# Clean up
488
520
RUN apt autoremove -y && \
489
521
apt autoclean && \
490
- ln -s /usr/local/cuda-11.2 /targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.2 /targets/x86_64-linux/lib/libcusolver.so.10 && \
522
+ ln -s /usr/local/cuda-11.7 /targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.7 /targets/x86_64-linux/lib/libcusolver.so.10 && \
491
523
sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
492
524
ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color && \
493
525
rm -rf /var/lib/apt/lists/* && \
0 commit comments