diff --git a/requirements.txt b/requirements.txt
index 4d63a57..d9748a6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -79,7 +79,7 @@ six==1.14.0
 tensorboard==1.15.0
 tensorflow==1.15.0
 tensorflow-estimator==1.15.1
-tensorflow-gpu==1.15.0
+tensorflow-gpu==2.6.4
 termcolor==1.1.0
 terminado==0.8.3
 testpath==0.4.4