diff --git a/requirements.txt b/requirements.txt
index 1f33175..0c50901 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 python_version == 3.6.12
-tensorflow_gpu == 1.14.0
+tensorflow_gpu == 2.6.4
 numpy == 1.19.5
 scipy == 1.5.4
 h5py == 3.1.0
diff --git a/setup.py b/setup.py
index 54b28d8..923f09e 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 REQUIRES_PYTHON = ">=3.6"
 
 REQUIRED = [
-    "tensorflow_gpu<1.15",
+    "tensorflow_gpu<2.7",
     "absl-py>=0.9",
     "appdirs",
     "dnspython",