Dockerfile

FROM tensorflow/tensorflow:2.11.1-gpu

LABEL description="Jupyter / Tensorflow 2.11.1 / CUDA 11.2 / Python 3.8.10 / Ubuntu 20.04.3 LTS" \
      maintainer="https://github.com/rlan/notebooks"

# Avoid configuring tzdata
ARG DEBIAN_FRONTEND=noninteractive

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

# 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", "--ip='*'", "--port=8888", "--no-browser", "--allow-root"]