Install rocm-smi package

Adds the rocm-smi package to get info about amdgpu. Signed-off-by: Enrique Belarte Luque <[email protected]>
containers · Apr 29, 2024 · d997b07 · d997b07
1 parent 5c705b2
commit d997b07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/amd/Containerfile b/training/amd/Containerfile
@@ -11,7 +11,7 @@ RUN source /etc/os-release \
         && crb enable ;\
     fi
 
-RUN dnf install -y rocm && dnf clean all
+RUN dnf install -y rocm-smi && dnf clean all
 
 ARG INSTRUCTLAB_IMAGE="quay.io/ai-lab/instructlab-amd:latest"