Merge pull request #105 from kundajelab/austintwang-tf-dependency-patch

Migrate from tensorflow-gpu to tensorflow
kundajelab · Jun 1, 2023 · 5a6a689 · 5a6a689
2 parents ac06299 + dab48b9
commit 5a6a689
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -6,7 +6,7 @@ pandas>=1.3.4
 pyfaidx==0.6.1
 scikit-learn>=1.1.2
 scipy>=1.4.1
-tensorflow-gpu==2.8.0 
+tensorflow==2.8.0 
 tensorflow-estimator==2.8.0 
 tensorflow-probability==0.15.0
 protobuf==3.20