diff --git a/requirements.txt b/requirements.txt
index e097538..3d6059b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -79,7 +79,7 @@ six==1.15.0
 tensorboard==1.14.0
 tensorflow==1.14.0
 tensorflow-estimator==1.14.0
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.6.4
 tensorflow-tensorboard==0.4.0
 termcolor==1.1.0
 terminado==0.8.3