Merge pull request #700 from dibryant/amd-rt

Add rocm runtimes to python 3.11 runtime-images folder
opendatahub-io · Sep 18, 2024 · 83ccdb5 · 83ccdb5
2 parents 7fffb1a + 7f43794
commit 83ccdb5
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/jupyter/datascience/ubi9-python-3.11/runtime-images/rocm-pytorch-ubi9-py311.json b/jupyter/datascience/ubi9-python-3.11/runtime-images/rocm-pytorch-ubi9-py311.json
@@ -0,0 +1,12 @@
+{
+  "display_name": "Pytorch with ROCm and Python 3.11 (UBI9)",
+  "metadata": {
+    "tags": [
+      "rocm-pytorch"
+    ],
+    "display_name": "Pytorch with ROCm and Python 3.11 (UBI9)",
+    "image_name": "quay.io/opendatahub/workbench-images@sha256:a2fa688aa519033eba09c915cdcd63640599881dc3efb693f9d44fe420a5f14d",
+    "pull_policy": "IfNotPresent"
+  },
+  "schema_name": "runtime-image"
+}
diff --git a/jupyter/datascience/ubi9-python-3.11/runtime-images/rocm-tensorflow-ubi9-py311.json b/jupyter/datascience/ubi9-python-3.11/runtime-images/rocm-tensorflow-ubi9-py311.json
@@ -0,0 +1,12 @@
+{
+"display_name": "TensorFlow with ROCm and Python 3.11 (UBI9)",
+"metadata": {
+  "tags": [
+    "rocm-tensorflow"
+  ],
+  "display_name": "TensorFlow with ROCm and Python 3.11 (UBI9)",
+  "image_name": "quay.io/opendatahub/workbench-images@sha256:bb57cb053656de834f50d7a76e5b8b823ce1ec9ea38fce69d2f307602f3c7632",
+  "pull_policy": "IfNotPresent"
+},
+"schema_name": "runtime-image"
+}