Update tensorflow-related packages to tf2.9 version.

PiperOrigin-RevId: 506015220
google · Jan 31, 2023 · ad16825 · ad16825
1 parent b403905
commit ad16825
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/install-gpu.sh b/install-gpu.sh
@@ -57,4 +57,4 @@ echo "$(pip --version)"
 # Install python packages used by DeepConsensus.
 ################################################################################
 python3 -m pip install --user -r requirements.txt
-python3 -m pip install --user "tensorflow-gpu>=2.11.0"
+python3 -m pip install --user "tensorflow-gpu==2.9.1"
diff --git a/install.sh b/install.sh
@@ -57,4 +57,4 @@ echo "$(pip --version)"
 # Install python packages used by DeepConsensus.
 ################################################################################
 python3 -m pip install --user -r requirements.txt
-python3 -m pip install --user "intel-tensorflow>=2.11.0"
+python3 -m pip install --user "intel-tensorflow==2.9.1"
diff --git a/requirements.txt b/requirements.txt
@@ -4,9 +4,9 @@
 # incompatible when flexible versioning is used.
 # DeepConsensus should be used in a dedicated python virtualenv.
 pandas==1.5.1
-tf-models-official==2.11.0
+tf-models-official==2.9.1
 pyyaml==5.4.1  # because of tf-models-official
 ml_collections==0.1.1
 absl-py==1.0.0
-keras==2.11.0
+keras==2.9.0
 pysam==0.19.0
diff --git a/setup.py b/setup.py
@@ -43,8 +43,8 @@
 REQUIREMENTS = (here / 'requirements.txt').read_text().splitlines()
 EXTRA_REQUIREMENTS = {
     # Use '==' to pin the versions.
-    'cpu': ['intel-tensorflow==2.11.0'],
-    'gpu': ['tensorflow-gpu==2.11.0']
+    'cpu': ['intel-tensorflow==2.9.1'],
+    'gpu': ['tensorflow-gpu==2.9.1']
 }