Dockerfile

# This builder downloads and converts the MNIST dataset to HDF5
FROM --platform=linux/amd64 mambaorg/micromamba:1.4.6-jammy AS builder

# RUN micromamba install -y -n base -c conda-forge numpy datasets h5py Pillow
# RUN micromamba env export -n base > /home/mambauser/environment_builder.yml

COPY environment_builder.yml /home/mambauser/environment_builder.yml
RUN micromamba install -y -f /home/mambauser/environment_builder.yml -n base

USER root
RUN apt-get update && apt-get install -y git
RUN echo "afb003abd0296f73cf572c61307be626c93012f6" > /backprop_tools_commit # because ARG does not invalidate the build cache
WORKDIR /
RUN git clone https://github.com/BackpropTools/BackpropTools
WORKDIR /BackpropTools
RUN git checkout $(cat /backprop_tools_commit)
ARG MAMBA_DOCKERFILE_ACTIVATE=1
WORKDIR /
RUN python3 /BackpropTools/examples/docker/00_basic_mnist/fetch_and_convert_mnist.py
USER mambauser

FROM --platform=linux/amd64 mambaorg/micromamba:1.4.6-jammy
ENV DEBIAN_FRONTEND=noninteractive
COPY --from=builder /home/mambauser/environment_builder.yml /environment_builder.yml

USER root
RUN apt-get update && apt-get install -y git
USER mambauser

# RUN micromamba install -y -n base -c conda-forge xeus-cling \
# xcanvas \
# hdf5 openblas \
# jupyterlab notebook jupyterlab_vim
# USER root
# RUN micromamba env export -n base > /environment.yml
# USER mambauser

# freezing env:
# ./run_bash.sh
# cp /environment.yml . && cp /environment_builder.yml .
# comment out the above
# comment in the below

COPY environment.yml /
RUN micromamba install -y -f /environment.yml -n base

USER root
COPY --from=builder /backprop_tools_commit /backprop_tools_commit
WORKDIR /
RUN git clone https://github.com/BackpropTools/BackpropTools
RUN mkdir -p /usr/local/include
RUN ln -s /BackpropTools/include/backprop_tools /usr/local/include/
WORKDIR /BackpropTools
RUN git checkout $(cat /backprop_tools_commit)
RUN git submodule update --init --recursive -- external/highfive
COPY --from=builder /mnist.hdf5 /data/mnist.hdf5
USER mambauser

WORKDIR /home/mambauser
RUN mkdir docs
COPY docs/*.ipynb ./docs/
COPY docs/images/ ./docs/images
WORKDIR docs

ENV LD_LIBRARY_PATH=/opt/conda/lib
ENV C_INCLUDE_PATH="/usr/local/include:/opt/conda/include:/BackpropTools/external/highfive/include"
ENV CPLUS_INCLUDE_PATH="/usr/local/include:/opt/conda/include:/BackpropTools/external/highfive/include"