Dockerize PyBaMM

Dockerfile

@@ -0,0 +1,36 @@
+FROM python:3.11-slim
+
+# Set the working directory
+WORKDIR /
+
+# Install the necessary dependencies
+RUN apt-get update && apt-get -y upgrade
+RUN apt-get install -y libopenblas-dev gcc gfortran graphviz git
+
+ENV CMAKE_C_COMPILER=/usr/bin/gcc
+ENV CMAKE_MAKE_PROGRAM=/usr/bin/make
+
+# Copy project files into the container
+RUN git clone https://github.com/pybamm-team/PyBaMM.git
+
+WORKDIR /PyBaMM/
+
+# Install PyBaMM
+RUN python -m pip install --upgrade pip
+RUN pip install -e ".[all]"
+
+ARG ODES
+ARG JAX
+
+
+RUN if [ "$ODES" = "true" ]; then \
+    apt-get install -y cmake && \
+    pip install wget && \
+    pybamm_install_odes; \
+    fi
+
+RUN if [ "$JAX" = "true" ]; then \
+    pip install -e ".[jax,all]";\
+    fi
+
+CMD ["/bin/bash"]