Bump up TensorFlow version in tests

- the latest to 2.15 - previous to 2.14.1 Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
NVIDIA · Nov 20, 2023 · 41d2e5c · 41d2e5c
1 parent 1cb4835
commit 41d2e5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -470,10 +470,10 @@ def get_pyvers_name(self, url, cuda_version):
                             "cupy-cuda11x"),
                 CudaPackage("tensorflow-gpu",
                             {"110": [
-                                PckgVer("2.12.1", python_min_ver="3.8", alias="tensorflow",
+                                PckgVer("2.14.1", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24",
                                                       "urllib3<2.0", "typing_extensions<4.6"]),
-                                PckgVer("2.13.0", python_min_ver="3.8", alias="tensorflow",
+                                PckgVer("2.15.0", python_min_ver="3.8", alias="tensorflow",
                                         dependencies=["protobuf<4", "numpy<1.24",
                                                       "urllib3<2.0", "typing_extensions<4.6"])]}),
                 CudaPackageExtraIndex("torch",