diff --git a/requirements.txt b/requirements.txt
index ebb13b9f..2d0f4bf9 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.6.4
diff --git a/setup.py b/setup.py
index 06258694..66339349 100644
--- a/setup.py
+++ b/setup.py
@@ -43,10 +43,10 @@
         'tensorflow-text==1.15.0rc0',
         'tfds-nightly',
         'tensor2tensor==1.15.0',
-        'tensorflow-gpu==1.15.2',
+        'tensorflow-gpu==2.6.4',
     ],
     extras_require={
-        'tensorflow': ['tensorflow-gpu==1.15.2'],
+        'tensorflow': ['tensorflow-gpu==2.6.4'],
     },
     classifiers=[
         'Development Status :: 4 - Beta',