diff --git a/requirements.txt b/requirements.txt
index ebb13b9f..8270af11 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,4 @@ sentencepiece
 tensorflow-text==1.15.0rc0
 tensor2tensor==1.15.0
 tensorflow-datasets==2.1.0
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.7.2