Dockerfile.pytorch2.6.0

FROM python:3.11-slim-bookworm

# Annotation spec: https://github.com/opencontainers/image-spec/blob/main/annotations.md
ARG DESCRIPTION="PyTorch 2.6.0 / Python 3.11.11 / python:3.11-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:6ed5bff4d7d377e2a27d9285553b8c21cfccc4f00881de1b24c9bc8d90016e82" \
      org.opencontainers.image.base.name="docker.io/library/python:3.11-slim-bookworm" \
      org.opencontainers.image.ref.name="python:3.11-slim-bookworm"

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 torchvision torchaudio --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"]