diff --git a/docker/rocm720.Dockerfile b/docker/rocm720.Dockerfile
index 6bcaed742c0c..9c75aed9ee0b 100644
--- a/docker/rocm720.Dockerfile
+++ b/docker/rocm720.Dockerfile
@@ -105,6 +105,9 @@ USER root
 RUN python -m pip install --upgrade pip && pip install setuptools_scm
 RUN apt-get purge -y sccache; python -m pip uninstall -y sccache; rm -f "$(which sccache)"
 
+# Install AMD SMI Python package from ROCm distribution
+RUN cd /opt/rocm/share/amd_smi && python3 -m pip install --no-cache-dir .
+
 WORKDIR /sgl-workspace
 
 # -----------------------
@@ -260,7 +263,7 @@ RUN /bin/bash -lc 'set -euo pipefail; \
       libgtest-dev libgmock-dev \
       libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev \
       python3 python3-dev python3-setuptools python3-pip python3-apt \
-      gcc libtinfo-dev zlib1g-dev libedit-dev libxml2-dev \
+      gcc libtinfo-dev zlib1g-dev libedit-dev libxml2-dev vim \
       cmake ninja-build pkg-config libstdc++6 software-properties-common \
   && rm -rf /var/lib/apt/lists/*; \
   \