From 6bd784e9c03a4e94274c10e18b3419ac77c69be2 Mon Sep 17 00:00:00 2001
From: yukirora <yuting.jiang@microsoft.com>
Date: Wed, 18 Sep 2024 14:52:57 +0000
Subject: [PATCH] add amd smi library install

---
 dockerfile/rocm6.2.x.dockerfile | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dockerfile/rocm6.2.x.dockerfile b/dockerfile/rocm6.2.x.dockerfile
index 5b48d1283..3fda1b7b5 100644
--- a/dockerfile/rocm6.2.x.dockerfile
+++ b/dockerfile/rocm6.2.x.dockerfile
@@ -148,6 +148,11 @@ RUN cd /opt/ &&  \
     .. && \
     make -j${NUM_MAKE_JOBS}
 
+# Install AMD SMI Python Library
+RUN apt install amd-smi-lib -y && \
+    cd /opt/rocm/share/amd_smi && \
+    python3 -m pip install .
+
 ENV PATH="/usr/local/mpi/bin:/opt/superbench/bin:/usr/local/bin/:/opt/rocm/hip/bin/:/opt/rocm/bin/:${PATH}" \
     LD_PRELOAD="/opt/rccl/build/librccl.so:$LD_PRELOAD" \
     LD_LIBRARY_PATH="/usr/local/mpi/lib:/usr/lib/x86_64-linux-gnu/:/usr/local/lib/:/opt/rocm/lib:${LD_LIBRARY_PATH}" \