proper version tensorflow

berkeley-dsep-infra · Aug 19, 2024 · e421e19 · e421e19
1 parent e6df54e
commit e421e19
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deployments/astro/image/environment.yml b/deployments/astro/image/environment.yml
@@ -39,7 +39,8 @@ dependencies:
 - scikit-learn==1.5.*
 - scikit-image==0.24.*
 - scipy==1.14.*
-- tensorflow-cpu==2.12.1
+- tensorflow-cpu==2.17.0
+- tensorflow==2.17.0
 - jax>=0.4.31
 - jaxlib>=0.4.31
 - cython==3.0.11