minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Clint
2026-02-22 21:32:14 -08:00
committed by GitHub
parent 2472e47d73
commit 3224836d8b
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/rocm720.Dockerfile
+4 -1
View File
@@ -105,6 +105,9 @@ USER root
RUN python -m pip install --upgrade pip && pip install setuptools_scm 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)" 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 WORKDIR /sgl-workspace
# ----------------------- # -----------------------
@@ -260,7 +263,7 @@ RUN /bin/bash -lc 'set -euo pipefail; \
libgtest-dev libgmock-dev \ libgtest-dev libgmock-dev \
libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev \ libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev \
python3 python3-dev python3-setuptools python3-pip python3-apt \ 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 \ cmake ninja-build pkg-config libstdc++6 software-properties-common \
&& rm -rf /var/lib/apt/lists/*; \ && rm -rf /var/lib/apt/lists/*; \
\ \