FROM quay.io/jupyter/minimal-notebook:latest AS builder

ARG PYTORCH_VARIANT=cpu

USER root

RUN apt-get update && \

apt-get install -y --no-install-recommends \

build-essential \

gcc \

g++ \

libffi-dev \

libgl1 && \

apt-get clean && \

rm -rf /var/lib/apt/lists/*

USER ${NB_UID}

COPY docker/environment.yml /tmp/environment.yml

RUN mamba env update -n base -f /tmp/environment.yml && \

# Force remove any existing PyTorch installations first

pip uninstall -y torch torchvision && \

# Install PyTorch packages without cache - conditionally based on variant

if [ "$PYTORCH_VARIANT" = "cpu" ]; then \

echo "Installing CPU-only PyTorch" && \

pip install --no-cache-dir --force-reinstall torch torchvision --index-url https://download.pytorch.org/whl/cpu; \

else \

echo "Installing CUDA-enabled PyTorch" && \

pip install --no-cache-dir --force-reinstall torch torchvision; \

fi && \

# Clean up all package caches to reduce image size

mamba clean --all -f -y && \

# Remove pip cache

rm -rf ~/.cache/pip && \

fix-permissions "${CONDA_DIR}" && \

fix-permissions "/home/${NB_USER}"

FROM quay.io/jupyter/minimal-notebook:latest

ARG PYTORCH_VARIANT=cpu

LABEL org.opencontainers.image.title="Python Tutorial"

LABEL org.opencontainers.image.description="A containerized Python tutorial environment with Jupyter Lab."

LABEL org.opencontainers.image.authors="Empa Scientific IT <scientificit@empa.ch>"

LABEL org.opencontainers.image.url="https://github.com/empa-scientific-it/python-tutorial"

LABEL org.opencontainers.image.source="https://github.com/empa-scientific-it/python-tutorial"

LABEL org.opencontainers.image.version="1.0.0"

LABEL org.opencontainers.image.licenses="MIT"

LABEL org.opencontainers.image.variant="pytorch-${PYTORCH_VARIANT}"

USER root

RUN apt-get update && \

apt-get install -y --no-install-recommends \

libgl1 && \

apt-get clean && \

rm -rf /var/lib/apt/lists/*

COPY docker/jupyter_server_config.py /opt/conda/etc/jupyter/

RUN mkdir -p /opt/conda/share/jupyter/lab/settings

COPY docker/overrides.json /opt/conda/share/jupyter/lab/settings/

RUN mkdir -p /opt/conda/etc/ipython

COPY docker/ipython_config.py /opt/conda/etc/ipython/

USER ${NB_UID}

COPY --from=builder ${CONDA_DIR} ${CONDA_DIR}

COPY --from=builder --chown=${NB_UID}:${NB_GID} /home/${NB_USER} /home/${NB_USER}

WORKDIR /home/${NB_USER}

ENTRYPOINT ["tini", "-g", "--", "start.sh"]