jupyter linux docker gpu

build.sh

#!/usr/bin/bash
docker build -t jupyter-lab-gpu .

Dockerfile

FROM tensorflow/tensorflow:latest-gpu-py3-jupyter

# RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - \
#     && apt-get install -y nodejs \
#     && rm -rf /var/lib/apt/lists/*

RUN pip install -U \
    jupyter \
    jupyterlab \
    pandas

#    jupyter_tensorboard \

#RUN jupyter labextension install \
#    jupyterlab_tensorboard

ENV SHELL=/bin/bash

CMD ["bash", "-c", \
    "source /etc/bash.bashrc && jupyter lab --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root"]

start.sh

#!/usr/bin/bash

#IMG=jupyter/tensorflow-notebook:latest
#IMG=tensorflow/tensorflow:latest-gpu-py3-jupyter
#IMG=cschranz/gpu-jupyter:latest
IMG=jupyter-lab-gpu

#WORKDIR=/home/jovyan
WORKDIR=/tf


docker run \
    --rm \
    -it \
    --gpus all \
    -p 8888:8888 \
    -p 6006:6006 \
    -e JUPYTER_ENABLE_LAB=yes \
    -e GRANT_SUDO=yes \
    --user root \
    -v "$PWD:$WORKDIR" \
    --name jupyter \
    $IMG