diff --git a/requirements.txt b/requirements.txt
index cff22f35bc60..d3a9447d6b8d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,7 @@ fakeredis==1.0.3
 pymongo==3.8.0
 numpy==1.16.3
 scipy==1.2.1
-tensorflow==1.13.1
+tensorflow==2.5.2
 apscheduler==3.6.0
 tqdm==4.31.0
 networkx==2.3
diff --git a/setup.py b/setup.py
index 7efa8ec2e8cf..efd9acbeaa7e 100644
--- a/setup.py
+++ b/setup.py
@@ -37,7 +37,7 @@
     "pymongo~=3.8",
     "numpy~=1.16",
     "scipy~=1.2",
-    "tensorflow~=1.13.0",
+    "tensorflow>=1.13,<2.6",
     "apscheduler~=3.0",
     "tqdm~=4.0",
     "networkx~=2.3",