From bb0c87241cb69f325cdd0be27ab2035dc298205a Mon Sep 17 00:00:00 2001
From: Michael Thiel <mt255459@fh-muenster.de>
Date: Fri, 7 Mar 2025 10:46:08 +0000
Subject: [PATCH] added jaxlib=cuda

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 36882de..9ea4d08 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -16,7 +16,7 @@ RUN mamba install -c conda-forge myst-nb
 RUN mamba install -c conda-forge ipympl
 RUN mamba install -c conda-forge ipywidgets
 
-RUN mamba install -c conda-forge jax
+RUN mamba install -c conda-forge "jaxlib=*=*cuda*" jax
 
 RUN mamba install -c conda-forge keras
 RUN mamba install -c conda-forge control
-- 
GitLab