diff --git a/requirements.txt b/requirements.txt index a681b299fe..6b673d7a63 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,7 +8,7 @@ redis==3.3.5 pymongo[tls,srv]==3.8.0 numpy==1.16.3 scipy==1.2.1 -tensorflow-cpu==1.15.0 +tensorflow-cpu==2.7.2 absl-py>=0.8.0 # setuptools comes from tensorboard requirement: # https://github.com/tensorflow/tensorboard/blob/1.14/tensorboard/pip_package/setup.py#L33 diff --git a/setup.py b/setup.py index d227020c47..a8ef07718c 100644 --- a/setup.py +++ b/setup.py @@ -37,7 +37,7 @@ "pymongo[tls,srv]~=3.8", "numpy~=1.16", "scipy~=1.2", - "tensorflow-cpu~=1.15.0", + "tensorflow-cpu>=1.15,<2.8", # absl is a tensorflow dependency, but produces double logging before 0.8 # should be removed once tensorflow requires absl > 0.8 on its own "absl-py>=0.8.0",