diff --git a/setup.py b/setup.py
index d5ce38d..e14d88b 100644
--- a/setup.py
+++ b/setup.py
@@ -62,7 +62,7 @@
         'numpy',
         'scipy',
         "scikit-learn",
-        "tensorflow==1.15",
+        "tensorflow==2.11.1",
         "torch>=1.2",
         "keras>=2.3",
         "matplotlib",