基于docker-deepo安装定制化jupyterlab+ssh访问

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

本博客基于深度学习deepo镜像安装jupyterlab及其debugger、git插件，并使容器支持ssh远程访问。
提示：以下是本篇文章正文内容，下面案例可供参考

一、环境配置

该部分所有内容可参考：
Workstation Setup for Docker with the New NVIDIA Container Toolkit

1. docker安装

参考：Install Docker Engine on Ubuntu

为了支持docker run 的–gpus参数，需要安装$\ge$ 19的。本人安装的是 19.03.13

2. nvidia驱动安装

参考：
Ubuntu PPA安装NVIDIA-drivers

目前最新的deepo镜像中cuda版本为10.1，需要到nvidia驱动版本查询查询一下自己的显卡适配的驱动版本。
不建议采用PPA以外方式安装驱动，坑太多。

3. 安装NVIDIA Container Toolkit

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list


sudo apt-get update

sudo apt-get install nvidia-container-toolkit


sudo systemctl restart docker

二、基于docker-deepo安装定制化jupyterlab

1. 目录结构

• jupyterlab_deepo
  • Dockerfile
  • run.sh

2. Dockerfile

FROM ufoym/deepo:latest
RUN apt-get update && curl -sL https://deb.nodesource.com/setup_12.x | bash - && apt-get install -y nodejs
RUN apt-get update && apt-get install -y openssh-server curl
RUN mkdir -p /root/.ssh
RUN sed -ri 's/^PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN sed -ri 's/^#PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN echo "root:root" | chpasswd
RUN python -m pip install jupyterlab ipympl

# git extension
RUN jupyter labextension install  @jupyterlab/git @jupyterlab/debugger && pip install jupyterlab-git && jupyter serverextension enable --py jupyterlab_git
# debuger
RUN jupyter labextension install @jupyterlab/debugger && python -m pip install xeus-python ptvsd

ENV NOTEBOOK_APP_TOKEN=passwd NOTEBOOK_DIR=/home
EXPOSE 8888 22 6006
ADD run.sh /run.sh
RUN chmod 755 run.sh
#CMD jupyter lab --ip=0.0.0.0 --no-browser --allow-root --NotebookApp.token=${NOTEBOOK_APP_TOKEN} --notebook-dir=${NOTEBOOK_DIR}
CMD ["/run.sh"]

3. run.sh

#! /bin/bash
service ssh start
jupyter lab --ip=0.0.0.0 --no-browser --allow-root --NotebookApp.token=${NOTEBOOK_APP_TOKEN} --notebook-dir=${NOTEBOOK_DIR}

4. 镜像构建

docker build -t jupyterlab .build/

5. 启动容器

docker run --rm --gpus all -it -p 8848:8888 -p 2232:22 -v /home/jupyter_data:/home -e GRANT_SUDO=yes -e JUPYTER_ENABLE_LAB=yes --user root  --name gpu-jupyter_1 jupyterlab:latest 

其中/home/jupyter_data:/home为宿主机代码目录

6. 访问容器

1. 远程ssh访问ssh root@ip -p 2232，密码root
2. 浏览器访问ip:8848，密码:passwd