diff --git a/requirements.txt b/requirements.txt
index 05ebc36..f64ca22 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -32,7 +32,7 @@ six==1.14.0
 tensorboard==2.1.0
 tensorboardX==2.0
 tensorflow-estimator==2.1.0
-tensorflow-gpu==2.1.0
+tensorflow-gpu==2.7.2
 termcolor==1.1.0
 torch==1.4.0
 torchfile==0.1.0