Dockerfile

FROM tensorflow/tensorflow:2.3.4-gpu

LABEL description="Jupyter / Tensorflow 2.3.4 / CUDA 10.1.243-1 / Python 3.6.9 / Ubuntu 18.04.5 LTS" \
      maintainer="https://github.com/rlan/notebooks"

# Avoid configuring tzdata
ARG DEBIAN_FRONTEND=noninteractive

# Build-time metadata as defined at http://label-schema.org
ARG BUILD_DATE
ARG VCS_REF
ARG VERSION
LABEL org.label-schema.build-date=$BUILD_DATE \
      org.label-schema.vcs-ref=$VCS_REF \
      org.label-schema.vcs-url="https://github.com/rlan/notebooks" \
      org.label-schema.version=$VERSION \
      org.label-schema.schema-version="1.0"

RUN apt-get -qq update \
    && apt-get -qq install -y --no-install-recommends \
        python3-pandas \
    && apt-get -qq clean \
    && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

RUN pip --no-cache-dir install -q --upgrade pip setuptools wheel
        
RUN pip --no-cache-dir install -q --upgrade \
        jupyterlab \
        matplotlib \
        nltk \
        opencv-python-headless \
        pillow \
        scikit-learn \
        scikit-image \
        seaborn \
        tornado

# jupyter-tensorboard need tornado first
RUN pip --no-cache-dir install -q --upgrade \
        jupyter-tensorboard

# Set up our notebook config.
COPY jupyter_notebook_config.py /root/.jupyter/

# Jupyter has issues with being run directly:
#   https://github.com/ipython/ipython/issues/7062
# We just add a little wrapper script.
COPY run_jupyter.sh /

# TensorBoard
EXPOSE 6006
# IPython
EXPOSE 8888

WORKDIR /notebooks

CMD ["/run_jupyter.sh", "--allow-root"]