diff --git a/requirements.txt b/requirements.txt
index 82444da..6d96b03 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,7 @@ numpy < 1.19.0
 pylint == 2.6.0
 fire >= 0.3.1
 jupyter >= 1.0.0
-tensorflow == 2.3.1
+tensorflow == 2.6.4
 pyarrow <0.18,>=0.17
 frozendict == 1.2
 dataclasses >= 0.5;python_version<"3.7"
diff --git a/setup.py b/setup.py
index 6013f68..a38bfc2 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@
     "pylint >= 2.5.3",
     "pytz >= 2020.1",
     "python-dateutil",
-    "tensorflow == 2.3.1",
+    "tensorflow == 2.6.4",
     "tensorflow_transform == 0.24.1",
 ]