RHAIENG-1512: fix(ROCm/TensorFlow) repository URL to install ROCm 6.3.4 as in AIPCC base image (#2595)

Author: daniellutz

jupyter/rocm/tensorflow/ubi9-python-3.12/build-args/rocm.conf

@@ -1 +1 @@
-BASE_IMAGE=quay.io/opendatahub/odh-base-image-rocm-py312-ubi9:v6.4
+BASE_IMAGE=quay.io/opendatahub/odh-base-image-rocm-py312-ubi9:v6.3

jupyter/rocm/tensorflow/ubi9-python-3.12/pyproject.toml

@@ -8,15 +8,17 @@ dependencies = [
 
     # ROCm TensorFlow packages
     # https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html#using-a-wheels-package
-    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/tensorflow_rocm-2.18.1-cp312-cp312-manylinux_2_28_x86_64.whl#sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598",
-    "tensorboard~=2.18.0", # tensorflow-rocm==2.18.1 is available and tensorflow-rocm==2.18.1 depends on tensorboard>=2.18,<2.19
+    # TODO(jdanek): consider adding #sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598
+    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3/tensorflow_rocm-2.17.0-cp312-cp312-manylinux_2_28_x86_64.whl",
+    # tensorflow-rocm==2.x needs tensorboard>=2.x,<2.x+1
+    "tensorboard~=2.17.0",
 
     # Datascience and useful extensions
     "boto3~=1.40.52",
     "kafka-python-ng~=2.2.3",
     "kfp~=2.14.6",
     "matplotlib~=3.10.7",
-    "numpy~=2.0.2",
+    "numpy~=1.26.4",
     "pandas~=2.3.3",
     "plotly~=6.3.1",
     "scikit-learn~=1.7.2",

manifests/base/jupyter-rocm-tensorflow-notebook-imagestream.yaml

@@ -20,19 +20,19 @@ spec:
         # language=json
         opendatahub.io/notebook-software: |
           [
-            {"name": "ROCm", "version": "v6.4"},
+            {"name": "ROCm", "version": "v6.3"},
             {"name": "Python", "version": "v3.12"},
-            {"name": "TensorFlow-ROCm", "version": "2.18"}
+            {"name": "TensorFlow-ROCm", "version": "2.17"}
           ]
         # language=json
         opendatahub.io/notebook-python-dependencies: |
           [
             {"name": "JupyterLab","version": "4.4"},
-            {"name": "TensorFlow-ROCm", "version": "2.18"},
-            {"name": "Tensorboard", "version": "2.18"},
+            {"name": "TensorFlow-ROCm", "version": "2.17"},
+            {"name": "Tensorboard", "version": "2.17"},
             {"name": "Kafka-Python-ng", "version": "2.2"},
             {"name": "Matplotlib", "version": "3.10"},
-            {"name": "Numpy", "version": "2.0"},
+            {"name": "Numpy", "version": "1.26"},
             {"name": "Pandas", "version": "2.3"},
             {"name": "Scikit-learn", "version": "1.7"},
             {"name": "Scipy", "version": "1.16"},

runtimes/rocm-tensorflow/ubi9-python-3.12/build-args/rocm.conf

@@ -1 +1 @@
-BASE_IMAGE=quay.io/opendatahub/odh-base-image-rocm-py312-ubi9:v6.2
+BASE_IMAGE=quay.io/opendatahub/odh-base-image-rocm-py312-ubi9:v6.3

runtimes/rocm-tensorflow/ubi9-python-3.12/pyproject.toml

@@ -8,14 +8,16 @@ dependencies = [
 
     # ROCm TensorFlow packages
     # https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html#using-a-wheels-package
-    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/tensorflow_rocm-2.18.1-cp312-cp312-manylinux_2_28_x86_64.whl#sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598",
-    "tensorboard~=2.18.0", # tensorflow-rocm==2.18.1 is available and tensorflow-rocm==2.18.1 depends on tensorboard>=2.18,<2.19
+    # TODO(jdanek): consider adding #sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598
+    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3/tensorflow_rocm-2.17.0-cp312-cp312-manylinux_2_28_x86_64.whl",
+    # tensorflow-rocm==2.x needs tensorboard>=2.x,<2.x+1
+    "tensorboard~=2.17.0",
 
     # Datascience and useful extensions
     "boto3~=1.40.52",
     "kafka-python-ng~=2.2.3",
     "matplotlib~=3.10.7",
-    "numpy~=2.0.2",
+    "numpy~=1.26.4",
     "pandas~=2.3.3",
     "plotly~=6.3.1",
     "scikit-learn~=1.7.2",

tests/test_main.py

@@ -225,13 +225,12 @@ class VersionData:
             "Numpy",
             (
                 "1.26",  # trustyai 0.6.2 depends on numpy~=1.26.4
-                "2.0",  # for tensorflow rocm
                 "2.1",  # for tensorflow cuda
                 "2.2",  # for python 3.11 n-1 images
                 "2.3",  # this is our latest where possible
             ),
         ),
-        ("Tensorboard", ("2.20", "2.18")),
+        ("Tensorboard", ("2.20", "2.17")),
         ("PyTorch", ("2.6", "2.7")),
     )
 
@@ -277,7 +276,7 @@ def test_image_pyprojects_version_alignment(subtests: pytest_subtests.plugin.Sub
         # ("package name", ("allowed specifier 1", "allowed specifier 2", ...))
         ("setuptools", ("~=80.9.0", "==80.9.0")),
         ("wheel", ("==0.45.1", "~=0.45.1")),
-        ("tensorboard", ("~=2.18.0", "~=2.20.0")),
+        ("tensorboard", ("~=2.17.0", "~=2.20.0")),
         ("torch", ("==2.7.1", "==2.7.1+cu128", "==2.7.1+rocm6.2.4")),
         ("torchvision", ("==0.22.1", "~=0.22.1", "==0.22.1+cu128", "==0.22.1+rocm6.2.4")),
         (
@@ -288,7 +287,6 @@ def test_image_pyprojects_version_alignment(subtests: pytest_subtests.plugin.Sub
             "numpy",
             (
                 "~=1.26.4",  # trustyai 0.6.2 depends on numpy~=1.26.4
-                "~=2.0.2",  # for tensorflow rocm
                 "~=2.1.3",
                 "~=2.2.6",
                 "~=2.3.4",  # for tensorflow cuda and latest possible