modify tensorflow version

djmartingale · Aug 27, 2021 · d65ea15 · d65ea15
1 parent c8bc675
commit d65ea15
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/fate_client/pyproject.toml b/python/fate_client/pyproject.toml
@@ -34,7 +34,7 @@ requests = ">=2.24.0"
 click = "^7.1.2"
 "ruamel.yaml" = "^0.16.10"
 loguru = "^0.5.1"
-tensorflow = ">=2.4.1"
+tensorflow = "==2.3.4"
 torch = "1.4.0"
 flask = "^1.0.2"
 setuptools = "^50.0"

diff --git a/python/fate_client/setup.py b/python/fate_client/setup.py
@@ -38,7 +38,7 @@
     "requests_toolbelt>=0.9.1,<0.10.0",
     "ruamel.yaml>=0.16.10,<0.17.0",
     "setuptools>=50.0,<51.0",
-    "tensorflow>=2.4.1",
+    "tensorflow==2.3.4",
     "torch",
 ]
 

diff --git a/python/requirements.txt b/python/requirements.txt
@@ -20,7 +20,7 @@ requests_toolbelt==0.9.1
 ruamel-yaml==0.16.10
 scikit-learn==0.19.2
 scipy==1.1.0
-tensorflow>=2.4.1
+tensorflow==2.3.4
 torch==1.4.0
 torchvision==0.5.0
 Werkzeug==0.15.3