Re-instate tf2.2 minimum version (2.3 not on Conda for Linux)

o2team · Dec 20, 2020 · 9182c56 · 9182c56
1 parent 7c296e6
commit 9182c56
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/lib/cli/launcher.py b/lib/cli/launcher.py
@@ -53,9 +53,9 @@ def _test_for_tf_version(self):
         Raises
         ------
         FaceswapError
-            If Tensorflow is not found, or is not between versions 2.3 and 2.3
+            If Tensorflow is not found, or is not between versions 2.2 and 2.3
         """
-        min_ver = 2.3
+        min_ver = 2.2
         max_ver = 2.3
         try:
             # Ensure tensorflow doesn't pin all threads to one core when using Math Kernel Library

diff --git a/requirements_amd.txt b/requirements_amd.txt
@@ -1,3 +1,3 @@
 -r _requirements_base.txt
-tensorflow>=2.3.0,<2.4.0
+tensorflow>=2.2.0,<2.4.0
 plaidml-keras==0.7.0
diff --git a/requirements_cpu.txt b/requirements_cpu.txt
@@ -1,2 +1,2 @@
 -r _requirements_base.txt
-tensorflow>=2.3.0,<2.4.0
+tensorflow>=2.2.0,<2.4.0
diff --git a/requirements_nvidia.txt b/requirements_nvidia.txt
@@ -1,2 +1,2 @@
 -r _requirements_base.txt
-tensorflow-gpu>=2.3.0,<2.4.0
+tensorflow-gpu>=2.2.0,<2.4.0
diff --git a/setup.py b/setup.py
@@ -16,7 +16,7 @@
 
 INSTALL_FAILED = False
 # Revisions of tensorflow GPU and cuda/cudnn requirements
-TENSORFLOW_REQUIREMENTS = {">=2.3.0,<2.4.0": ["10.1", "7.6"]}
+TENSORFLOW_REQUIREMENTS = {">=2.2.0,<2.4.0": ["10.1", "7.6"]}
 # Mapping of Python packages to their conda names if different from pip or in non-default channel
 CONDA_MAPPING = {
     # "opencv-python": ("opencv", "conda-forge"),  # Periodic issues with conda-forge opencv