diff --git a/machine-learning/requirements.txt b/machine-learning/requirements.txt
index 7829b01..f746f1d 100644
--- a/machine-learning/requirements.txt
+++ b/machine-learning/requirements.txt
@@ -6,7 +6,7 @@ imblearn
 configparser
 joblib
 keras
-tensorflow-gpu
+tensorflow
 matplotlib
 scipy
 keras-metrics