diff --git a/setup.py b/setup.py
index 6bf65bc..49abc5c 100644
--- a/setup.py
+++ b/setup.py
@@ -67,7 +67,7 @@
         'sklearn==0.0',
         'tensorboard==2.2.2',
         'tensorboard-plugin-wit==1.6.0.post3',
-        'tensorflow==2.5.1',
+        'tensorflow==2.6.4',
         'tensorflow-estimator==2.2.0',
         'termcolor==1.1.0',
         'threadpoolctl==2.1.0',