Merge pull request #366 from enriquebelarte/amd-rocm-smi

Install rocm-smi package
containers · Apr 29, 2024 · d73db8d · d73db8d
2 parents 6b218d8 + d997b07
commit d73db8d
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"