diff --git a/requirements.txt b/requirements.txt
index 74487b7..6c37552 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -22,7 +22,7 @@ scipy==1.6.3
 six==1.12.0
 tensorboard==2.0.0
 tensorflow-estimator==2.0.0
-tensorflow-gpu==2.0.0
+tensorflow-gpu==2.4.0
 termcolor==1.1.0
 Werkzeug==0.16.0
 wrapt==1.11.2