Update RHOAI manifests to include universal workbench image configuration

Signed-off-by: kramaranya <kramaranya15@gmail.com>

Author: kramaranya

manifests/rhoai/kustomization.yaml

@@ -26,6 +26,19 @@ replacements:
       name: kubeflow-trainer-controller-manager
     fieldPaths:
     - spec.template.spec.containers.0.image
+- source:
+    kind: ConfigMap
+    name: rhoai-config
+    version: v1
+    fieldPath: data.odh-kubeflow-trainer-universal-workbench-image
+  targets:
+  - select:
+      group: image.openshift.io
+      version: v1
+      kind: ImageStream
+      name: training-hub-universal-cuda
+    fieldPaths:
+    - spec.tags.0.from.name
 
 # Labels to add to all resources and selectors.
 labels:
@@ -41,6 +54,7 @@ resources:
 - ../base/webhook
 # - ../third-party/jobset #uncomment if jobset should be bundled under kubeflow trainer controller manager
 - runtimes
+- training-hub-universal-cuda-imagestream.yaml
 - kubeflow-training-roles.yaml
 - monitor.yaml
 

manifests/rhoai/params.env

@@ -1 +1,2 @@
 odh-kubeflow-trainer-controller-image=quay.io/opendatahub/trainer:v2.1.0
+odh-kubeflow-trainer-universal-workbench-image=quay.io/mstoklus/training-image:runtime-image-training-hub30

manifests/rhoai/training-hub-universal-cuda-imagestream.yaml

@@ -0,0 +1,36 @@
+apiVersion: image.openshift.io/v1
+kind: ImageStream
+metadata:
+  name: training-hub-universal-cuda
+  labels:
+    opendatahub.io/notebook-image: "true"
+  annotations:
+    opendatahub.io/notebook-image-url: "https://github.com/opendatahub-io/trainer"
+    opendatahub.io/notebook-image-name: "Training Hub Universal (CUDA, Python 3.12)"
+    opendatahub.io/notebook-image-desc: "Universal Training Hub workbench image for CUDA-based fine-tuning and distributed training."
+    opendatahub.io/notebook-image-order: "1"
+    opendatahub.io/recommended-accelerators: '["nvidia.com/gpu"]'
+spec:
+  lookupPolicy:
+    local: true
+  tags:
+    - name: latest
+      annotations:
+        opendatahub.io/notebook-software: |
+          [
+            {"name": "CUDA", "version": "12.6"},
+            {"name": "Python", "version": "v3.12"},
+            {"name": "Training Hub", "version": "v0.3.0"}
+          ]
+        opendatahub.io/notebook-python-dependencies: |
+          [
+            {"name": "JupyterLab", "version": "4.4"}
+          ]
+        openshift.io/imported-from: quay.io/mstoklus/training-image
+      from:
+        kind: DockerImage
+        name: odh-kubeflow-trainer-universal-workbench-image
+      importPolicy:
+        importMode: Legacy
+      referencePolicy:
+        type: Source