Changed ROCm packages per level

ai-dock · Jun 16, 2024 · 368f4cc · 368f4cc
1 parent e92583e
commit 368f4cc
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/amd.sh b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/amd.sh
@@ -30,16 +30,18 @@ apt-get update
 
 if [[ "${ROCM_LEVEL}" == "core" ]]; then
     $APT_INSTALL rocm-core \
-                 rocm-opencl-runtime
+                 rocminfo
+
 
 elif [[ "${ROCM_LEVEL}" == "runtime" ]]; then
-    $APT_INSTALL rocm-dev
+    $APT_INSTALL hip-runtime-amd \
+                 rocm-opencl-runtime \
+                 miopen-hip \
+                 rocblas \
+                 rocfft
 
 elif [[ "${ROCM_LEVEL}" == "devel" ]]; then
-    $APT_INSTALL rocm-libs \
-                 rocm-opencl-sdk \
-                 rocm-hip-sdk \
-                 rocm-ml-sdk
+    $APT_INSTALL rocm
 
 else
     printf "No valid ROCM_LEVEL specified\n" >&2