Added the [and-gpu] flag to tensorflow to hopefully utilize the cluster gpu

2025-09-01 18:02:41 +02:00
parent 1e85d5f3e2
commit 4f4e9244ab
1 changed files with 1 additions and 1 deletions
@@ -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