diff --git a/keras_implementation/requirements.txt b/keras_implementation/requirements.txt
index 2062b55..7fed34c 100644
--- a/keras_implementation/requirements.txt
+++ b/keras_implementation/requirements.txt
@@ -8,6 +8,6 @@ scikit-learn==0.21.1
 matplotlib==3.1.0
 keras==2.2.4
 keras-gpu==2.2.4
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.4.0
 bokeh==1.1.0
 pydot==1.4.1