Move both ML frameworks from vanilla to firedrake

firedrakeproject · Sep 23, 2024 · e426715 · e426715
1 parent 9ed9765
commit e426715
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/docker/Dockerfile.firedrake b/docker/Dockerfile.firedrake
@@ -11,14 +11,17 @@ WORKDIR /home/firedrake
 RUN bash -c "source firedrake/bin/activate; \
     firedrake-update \
         --documentation-dependencies \
+        --jax \
         --netgen \
         --slepc \
-        --install femlium \
+        --torch \
+        --install asQ \
         --install defcon \
         --install fascd \
+        --install femlium \
         --install gadopt \
         --install gusto \
         --install icepack \
         --install irksome \
-        --install asQ \
         --install thetis"
+
diff --git a/docker/Dockerfile.vanilla b/docker/Dockerfile.vanilla
@@ -12,8 +12,6 @@ RUN curl -O https://raw.githubusercontent.com/firedrakeproject/firedrake/master/
 RUN bash -c "python3 firedrake-install \
     --no-package-manager \
     --disable-ssh \
-    --torch \
-    --jax \
     --honour-petsc-dir \
     --mpicc=$MPICH_DIR/mpicc \
     --mpicxx=$MPICH_DIR/mpicxx \