Merge pull request #2 from magpowell/maggie2

requirements file and new base image
2i2c-org · May 7, 2024 · 434fb39 · 434fb39
2 parents b7083c3 + 33002a9
commit 434fb39
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,8 +1,7 @@
-FROM pangeo/pangeo-notebook:2023.10.03
+FROM pangeo/ml-notebook
 
-RUN pip install git+https://github.com/google-research/neuralgcm
-RUN pip install git+https://github.com/google-research/dinosaur
-RUN pip install jupyterlab==4.1.6 gcsfs==2024.3.1 matplotlib==3.8.4 julia==0.6.2
+COPY requirements.txt requirements.txt
+RUN pip install -r requirements.txt
 
 USER root
 ENV DEBIAN_FRONTEND=noninteractive

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,11 @@
+dinosaur@git+https://github.com/google-research/dinosaur
+dm-haiku
+gin-config
+jax
+jaxlib
+optax
+tensorflow_probability
+jupyterlab==4.1.6 
+gcsfs==2024.3.1 
+matplotlib==3.8.4 
+julia==0.6.2