diff --git a/AttentiveRanker/src/requirements.txt b/AttentiveRanker/src/requirements.txt
index 81d3454..969d024 100644
--- a/AttentiveRanker/src/requirements.txt
+++ b/AttentiveRanker/src/requirements.txt
@@ -16,7 +16,7 @@ scipy==1.3.1
 six==1.12.0
 tensorboard==1.13.1
 tensorflow-estimator==1.13.0
-tensorflow-gpu==1.13.1
+tensorflow-gpu==2.7.2
 termcolor==1.1.0
 tqdm==4.35.0
 Werkzeug==0.15.6