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

4 days ago · 4f4e9244ab
1 changed files with 1 additions and 1 deletions
--- 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