diff --git a/requirements.txt b/requirements.txt
index 9c366ee..46569d1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -128,7 +128,7 @@ systemd-python==234
 tensorboard==1.14.0
 tensorboardX==1.8
 tensorflow-estimator==1.14.0
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.6.4
 termcolor==1.1.0
 terminado==0.8.2
 testpath==0.4.2