Fix the minimal Python version for TF 2.14 and 2.15, adds 2.13 (#5221)

- adjusts minimal Python version for TF 2.14 and 2.15 to Python 3.9 - adds DALI built-in support for TF 2.13.1 as the last one that supports Python 3.8 Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · Dec 6, 2023 · 40759f4 · 40759f4
1 parent 830761c
commit 40759f4
Showing 1 changed file with 26 additions and 4 deletions.
diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -488,7 +488,7 @@ def get_pyvers_name(self, url, cuda_version):
         {
             "110": [
                 PckgVer(
-                    "2.14.1",
+                    "2.13.1",
                     python_min_ver="3.8",
                     alias="tensorflow",
                     dependencies=[
@@ -497,11 +497,22 @@ def get_pyvers_name(self, url, cuda_version):
                         "urllib3<2.0",
                         "typing_extensions<4.6",
                     ],
-                )
+                ),
+                PckgVer(
+                    "2.14.1",
+                    python_min_ver="3.9",
+                    alias="tensorflow",
+                    dependencies=[
+                        "protobuf<4",
+                        "numpy<1.24",
+                        "urllib3<2.0",
+                        "typing_extensions<4.6",
+                    ],
+                ),
             ],
             "120": [
                 PckgVer(
-                    "2.14.1",
+                    "2.13.1",
                     python_min_ver="3.8",
                     alias="tensorflow",
                     dependencies=[
@@ -511,9 +522,20 @@ def get_pyvers_name(self, url, cuda_version):
                         "typing_extensions<4.6",
                     ],
                 ),
+                PckgVer(
+                    "2.14.1",
+                    python_min_ver="3.9",
+                    alias="tensorflow",
+                    dependencies=[
+                        "protobuf<4",
+                        "numpy<1.24",
+                        "urllib3<2.0",
+                        "typing_extensions<4.6",
+                    ],
+                ),
                 PckgVer(
                     "2.15.0",
-                    python_min_ver="3.8",
+                    python_min_ver="3.9",
                     alias="tensorflow",
                     dependencies=[
                         "protobuf<4",