Update rocm7.2 Dockerfile to install amdsmi for QuickReduce Initialization (#19091)

This commit is contained in:
Clint
2026-02-22 23:32:14 -06:00
committed by GitHub
parent 2472e47d73
commit 3224836d8b

View File

@@ -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/*; \
\