changed tensorflow-cpu to tensorflow

mauvais2 · mauvais2 · commit 1f058ab5282b · 2025-05-30T08:50:27.000-07:00
diff --git a/pip/cpu_requirements.txt b/pip/cpu_requirements.txt
@@ -5,7 +5,7 @@
 # for LC developers: comment out pypi index url and use t$
 # --extra-index-url https://wci-repo.llnl.gov/repository/$
 
-tensorflow-cpu~=2.14.0
+tensorflow~=2.14.0
 
 torch==2.0.1
 torch_geometric
diff --git a/pip/docker_requirements.txt b/pip/docker_requirements.txt
@@ -5,7 +5,7 @@
 # for LC developers: comment out pypi index url and use t$
 # --extra-index-url https://wci-repo.llnl.gov/repository/$
 
-tensorflow-cpu~=2.14.0
+tensorflow~=2.14.0
 
 torch==2.0.1
 torch_geometric
diff --git a/pip/readthedocs_requirements.txt b/pip/readthedocs_requirements.txt
@@ -15,6 +15,6 @@ seaborn>=0.13.0
 setuptools
 sphinx
 sphinx_rtd_theme
-tensorflow-cpu
+tensorflow
 torch
 umap-learn
