diff --git a/setup.py b/setup.py
index ead6d77..f7c39b9 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
         "attrs",
         "multiprocess",
         "opencv-python",
-        "tensorflow==2.3.1",
+        "tensorflow==2.6.4",
         "torch",
     ],
     scripts=['cluster/holy_dannce_train.sh',