Add rocm/amd-smi-lib

ai-dock · Jun 26, 2024 · b89ed2f · b89ed2f
1 parent d7a24d8
commit b89ed2f
Showing 1 changed file with 9 additions and 4 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,21 +30,26 @@ apt-get update
 
 if [[ "${ROCM_LEVEL}" == "core" ]]; then
     $APT_INSTALL rocm-core \
-                 rocminfo
-
+                 rocminfo \
+                 amd-smi-lib \
+                 rocm-smi-lib
 
 elif [[ "${ROCM_LEVEL}" == "runtime" ]]; then
     $APT_INSTALL hip-runtime-amd \
                  rocm-opencl-runtime \
                  miopen-hip \
                  rocblas \
-                 rocfft
+                 rocfft \
+                 amd-smi-lib \
+                 rocm-smi-lib
 
 elif [[ "${ROCM_LEVEL}" == "devel" ]]; then
     $APT_INSTALL rocm-libs \
                  rocm-opencl-sdk \
                  rocm-hip-sdk \
-                 rocm-ml-sdk
+                 rocm-ml-sdk \
+                 amd-smi-lib \
+                 rocm-smi-lib
 
 else
     printf "No valid ROCM_LEVEL specified\n" >&2