Dockerfile.pytorch2.6.0-cuda12.6

FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime

# Annotation spec: https://github.com/opencontainers/image-spec/blob/main/annotations.md
ARG DESCRIPTION="PyTorch 2.6.0 / Python 3.11.11 / pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime"
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:f894dae26e1ee8557c544f9cfdb9dc011b1552bf3c1e656b422f2e221d380e40" \
      org.opencontainers.image.base.name="docker.io/pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime" \
      org.opencontainers.image.ref.name="pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime"

RUN pip --no-cache-dir install -q --upgrade \
    ipywidgets \
    jupyterlab \
    nltk \
    opencv-python-headless \
    pandas \
    pytorch-lightning \
    scikit-learn \
    scikit-image \
    seaborn \
    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"]