diff --git a/lib/cli/launcher.py b/lib/cli/launcher.py index e867a681db..03d9e006d0 100644 --- 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 index 957ea579ff..a1ea23f152 100644 --- 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 index f89f053dfb..ab0d3cb7fa 100644 --- 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 index 5323101b27..4f59f5171c 100644 --- 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 index 438021a73b..4efb5b780c 100755 --- 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