Dockerfile.pytorch2.2.2

FROM python:3.10-slim-bookworm

# Annotation spec: https://github.com/opencontainers/image-spec/blob/main/annotations.md
ARG DESCRIPTION="PyTorch 2.2.2 / python:3.10-slim-bookworm"
ARG CREATED
ARG URL
ARG SOURCE
ARG VERSION
ARG REVISION
ARG VENDOR
ARG TITLE
LABEL org.label-schema.build-date=${CREATED} \
      org.label-schema.url=${URL} \
      org.label-schema.vcs-url=${SOURCE} \
      org.label-schema.version=${VERSION} \
      org.label-schema.vcs-ref=${REVISION} \
      org.label-schema.vendor=${VENDOR} \
      org.label-schema.name=${TITLE} \
      org.label-schema.description=${DESCRIPTION} \
      org.label-schema.schema-version="1.0" \
      org.opencontainers.image.created=${CREATED} \
      org.opencontainers.image.url=${URL} \
      org.opencontainers.image.source=${SOURCE} \
      org.opencontainers.image.version=${VERSION} \
      org.opencontainers.image.revision=${REVISION} \
      org.opencontainers.image.vendor=${VENDOR} \
      org.opencontainers.image.title=${TITLE} \
      org.opencontainers.image.description=${DESCRIPTION} \
      org.opencontainers.image.base.digest="sha256:5e38f0e65bd16247b4c2ad294d388c74bd96da3b7800edc570d72f2e3a403707" \
      org.opencontainers.image.base.name="docker.io/library/python:3.10-slim-bookworm" \
      org.opencontainers.image.ref.name="python:3.10-slim-bookworm"

#ARG PYTHON=python3
ARG PIP=pip3

RUN ${PIP} --no-cache-dir install -q --upgrade \
    ipywidgets \
    jupyterlab \
    nltk \
    opencv-python-headless \
    pandas \
    scikit-learn \
    scikit-image \
    seaborn

# Install PyTorch
RUN ${PIP} --no-cache-dir install -q --upgrade \
    torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cpu

RUN ${PIP} --no-cache-dir install -q --upgrade \
    pytorch-lightning \
    tensorboard

# 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 /

# IPython
EXPOSE 8888

# Tensorboard
EXPOSE 6006

WORKDIR /notebooks

CMD ["/run_jupyter.sh", "--ip='*'", "--port=8888", "--no-browser", "--allow-root"]