@@ -14,7 +14,7 @@ ARG python=3.6
14
14
ENV PYTHON_VERSION=${python}
15
15
16
16
ENV PYTHONUNBUFFERED=1 \
17
- LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/local/cuda/lib64/stubs:/usr/ include/x86_64-linux-gnu" \
17
+ LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
18
18
PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/bin/mecab" \
19
19
LANG=C.UTF-8
20
20
@@ -127,7 +127,7 @@ RUN mkdir -p ${STAGE_DIR} && \
127
127
tar xzf ${STAGE_DIR}/nvidia-peer-memory_1.0.orig.tar.gz && \
128
128
cd ${STAGE_DIR}/nvidia-peer-memory-1.0 && \
129
129
dpkg-buildpackage -us -uc && \
130
- dpkg -i ${STAGE_DIR}/nvidia-peer-memory_1.0-8_all .deb
130
+ dpkg -i ${STAGE_DIR}/nvidia-peer-memory_1.0-9_all .deb
131
131
132
132
# Install CUDA-10.1 + cuDNN 7.6.0
133
133
RUN ln -s /usr/local/cuda-10.1 /usr/local/cuda && \
@@ -318,4 +318,18 @@ RUN apt autoclean && \
318
318
rm -rf /root/.cache && \
319
319
rm -rf /tmp/*
320
320
321
+ # Jupyter notebook extension
322
+ RUN mkdir -p /home/work/.jupyter/nbextension
323
+ WORKDIR /home/work/.jupyter/nbextension
324
+
325
+ RUN jupyter nbextensions_configurator enable && \
326
+ jupyter contrib nbextension install && \
327
+ jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
328
+ jupyter contrib nbextension install && \
329
+ jupyter serverextension enable --py jupyterlab --sys-prefix && \
330
+ jupyter labextension install @jupyter-widgets/jupyterlab-manager && \
331
+ # jupyter labextension install jupyterlab-nvdashboard && \
332
+ git clone https://github.com/lambdalisue/jupyter-vim-binding vim_binding && \
333
+ jupyter nbextension enable /home/work/.jupyter/nbextension/vim_binding/vim_binding
334
+
321
335
WORKDIR /home/work
0 commit comments