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