diff --git a/requirements.txt b/requirements.txt
index c27c0d2..11bec11 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -103,7 +103,7 @@ tb-nightly==1.14.0a20190603
 tensorboard==1.14.0
 tensorboardX==1.7
 tensorflow-estimator==1.13.0
-tensorflow-gpu==2.0.0b1
+tensorflow-gpu==2.5.1
 termcolor==1.1.0
 terminado==0.8.2
 testpath==0.4.2