This commit is contained in:
2026-03-15 22:50:28 +08:00
parent 085e4b76c5
commit 64e9dcc74a
2 changed files with 55 additions and 11 deletions
+3 -11
View File
@@ -46,17 +46,9 @@ RUN dnf -y install --setopt=install_weak_deps=False --nodocs \
WORKDIR /tmp
ARG ROCM_MAJOR_VER=7
ARG GFX=gfx120X-all
RUN set -euo pipefail; \
BASE="https://therock-nightly-tarball.s3.amazonaws.com"; \
PREFIX="therock-dist-linux-${GFX}-${ROCM_MAJOR_VER}"; \
KEY="$(maybe_proxy curl -s "${BASE}?list-type=2&prefix=${PREFIX}" \
| tr '<' '\n' \
| grep -o "therock-dist-linux-${GFX}-${ROCM_MAJOR_VER}\..*\.tar\.gz" \
| sort -V | tail -n1)"; \
echo "Downloading Latest Tarball: ${KEY}"; \
maybe_proxy aria2c -x 16 -s 16 -j 16 --file-allocation=none "${BASE}/${KEY}" -o therock.tar.gz; \
mkdir -p /opt/rocm; \
tar xzf therock.tar.gz -C /opt/rocm --strip-components=1; \
COPY therock.tar.gz /tmp/therock.tar.gz
RUN mkdir -p /opt/rocm && \
tar xzf therock.tar.gz -C /opt/rocm --strip-components=1 && \
rm therock.tar.gz
# 3. Configure Global ROCm Environment