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为宿主机代码目录