Changed ROCm packages per level

ai-dock · Jun 16, 2024 · d7a24d8 · d7a24d8
1 parent e92583e
commit d7a24d8
Showing 1 changed file with 7 additions and 2 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,10 +30,15 @@ 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 \