From 4f4e9244ab42eb1f0f388d07a671273faf18f268 Mon Sep 17 00:00:00 2001
From: Robert Rabbe <robert.rabbe@uni-saarland.de>
Date: Mon, 1 Sep 2025 18:02:41 +0200
Subject: [PATCH] Added the [and-gpu] flag to tensorflow to hopefully utilize
 the cluster gpu

---
 environment.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/environment.yaml b/environment.yaml
index 29a54e2..263a2b0 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -43,7 +43,7 @@ dependencies:
       - six==1.17.0
       - tensorboard==2.19.0
       - tensorboard-data-server==0.7.2
-      - tensorflow==2.19.0
+      - tensorflow[and-cuda]==2.19.0
       - tensorflow-io-gcs-filesystem==0.37.1
       - termcolor==3.1.0
       - threadpoolctl==3.6.0