From c3a0152e0b9a98784522d9fd3a4c6cc84f33b2e9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= <dtrifiro@redhat.com>
Date: Wed, 13 Nov 2024 13:06:01 +0100
Subject: [PATCH] Dockerfile.rocm.ubi: add amd-smi

---
 Dockerfile.rocm.ubi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.rocm.ubi b/Dockerfile.rocm.ubi
index 2607763cdb975..f070c9992af09 100644
--- a/Dockerfile.rocm.ubi
+++ b/Dockerfile.rocm.ubi
@@ -52,7 +52,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
         torch==2.6.0.dev20241107+rocm${version}\
         torchvision==0.20.0.dev20241107+rocm${version} && \
     # Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
-    microdnf install -y libdrm-amdgpu && \
+    microdnf install -y amd-smi libdrm-amdgpu && \
     microdnf clean all