Namespace
rocm
Image / Tagvllm:rocm6.4.1_vllm_0.9.0.1_20250605
Content Digestsha256:f48beeb3d72663a93c77211eb45273d564451447c097e060befa713d565fa36c
Details
2025-06-05 17:36:12 -0400
7.85 GB
Labels
-
maintainerdl.mlsedevops@amd.com
-
org.opencontainers.image.ref.nameubuntu
-
org.opencontainers.image.version22.04
Environment
DEBIAN_FRONTENDnoninteractive
HIP_FORCE_DEV_KERNARG1
LD_LIBRARY_PATH/opt/rocm/lib:/usr/local/lib:
PATH/opt/rocm/llvm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
PYTORCH_ROCM_ARCHgfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
PYTORCH_TUNABLEOP_ENABLED1
PYTORCH_TUNABLEOP_FILENAME/app/afo_tune_device_%d_full.csv
PYTORCH_TUNABLEOP_TUNING0
RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES1
ROCM_PATH/opt/rocm
SAFETENSORS_FAST_GPU1
TOKENIZERS_PARALLELISMfalse
Layers
[#000] sha256:215ed5a638430309375291c48a01872859a8dbf1331e54ba0af221918eb8ce2e - 0.35% (28.2 MB)
[#001] sha256:b527d63c6f9ab3193251345195c170d9a2fa87aea604a2fe5f75d7fcda028081 - 0.0% (240 Bytes)
[#002] sha256:d9ef619ff835cfb6de718d501e1175a83dffea975aac0b7e72efc90c24bcf53e - 50.98% (4 GB)
[#003] sha256:eaf29c7ab64b7d85cf6a1404ce251c4c405265c427324f9334682edda8dc9a5b - 0.0% (657 Bytes)
[#004] sha256:00cb7415963a8375a94d6915f1d1bdc962f48b0dbccbad469bb94df52f750901 - 0.0% (92 Bytes)
[#005] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#006] sha256:e27a581d0303bd425fe2d2cc021c0e5b73ffa48a4c2c1877cf94f5ea53391e8b - 1.34% (108 MB)
[#007] sha256:7635a9aa010fecf358edcb95fdd260e94c8b9202ca1b5032a07a6dbb154f62b4 - 0.82% (65.8 MB)
[#008] sha256:3f50ed0204a8e0f3e5ebd8e53d9aabced4298bff870da0100caa7cd3f186bcb3 - 10.92% (878 MB)
[#009] sha256:e258ab4ef1fd51157b6f0453f2c2bf12678736a85a9c2b1fdde4c28b20eed525 - 3.09% (248 MB)
[#010] sha256:34a21c53e8d8c38f1f9096661f5773c52fc5d97648c137cb3018f5b051f9baae - 0.01% (821 KB)
[#011] sha256:70a63a3668e988d114639b4a81d8f4d5cfaedf5b07920b22e6e6ad7e38c66f1e - 10.1% (812 MB)
[#012] sha256:f9c0fabead5c39eb01293cac78899d34c588118ccc708646c42e79471e2defbc - 9.97% (802 MB)
[#013] sha256:0f949bb83db9ef06a3603daa0713ec54a4739d6ba93549a3975b3281f37253c7 - 0.0% (379 Bytes)
[#014] sha256:b20457da04b5d128f9632ac5a461c6391300717449676298f63b56057ae111da - 0.76% (61.3 MB)
[#015] sha256:9be244ee7e674213e2b389d6f95fdfc07dbfa0ebbfb7dff58529c84cb5f2aaef - 1.35% (109 MB)
[#016] sha256:ed07410cf7c8320c640f2a59274fcfab921d82e5769b6995e764e92616a89c78 - 0.0% (182 KB)
[#017] sha256:f5acfd32e00548468621264281970394325acaa493d1ae1fefe5b96a9432223c - 0.0% (216 Bytes)
[#018] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#019] sha256:33ac45caea46c09826ed1f2727209e3b1d67d4ec4a8fe7f362b67c86357259a6 - 0.0% (1.98 KB)
[#020] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#021] sha256:a625a752079917fc9b4aa98a701ab14c2364110ecfd353b65425273c21266e0b - 0.2% (16.2 MB)
[#022] sha256:c82c6aae8349a01d71c04f3121b6ba2a32dbdbd3f5253a006a924624ec3363c4 - 10.1% (812 MB)
[#023] sha256:66144e3d2f001ac2fe4c15b875d70f64a1e0aa6260066a4038bac165da4ea049 - 0.0% (129 KB)
[#024] sha256:49656c7e9611b236c36c7d5fda25def824c0dc92aba3282a15cc51ba792d5684 - 0.0% (128 KB)
History
/bin/sh -c #(nop) ARG RELEASE
2025-04-28 09:44:40 UTC/bin/sh -c #(nop) ARG LAUNCHPAD_BUILD_ARCH
2025-04-28 09:44:40 UTC/bin/sh -c #(nop) LABEL org.opencontainers.image.ref.name=ubuntu
2025-04-28 09:44:40 UTC/bin/sh -c #(nop) LABEL org.opencontainers.image.version=22.04
2025-04-28 09:44:42 UTC/bin/sh -c #(nop) ADD file:59e67123ba6a5d9eea9813e7b2a767696f767c15c5b23c61c4d5bd6ba6fa9ac6 in /
2025-04-28 09:44:42 UTC/bin/sh -c #(nop) CMD ["/bin/bash"]
2025-05-22 15:52:30 UTC (buildkit.dockerfile.v0)LABEL maintainer=dl.mlsedevops@amd.com
2025-05-22 15:52:30 UTC (buildkit.dockerfile.v0)ARG ROCM_VERSION=5.3
2025-05-22 15:52:30 UTC (buildkit.dockerfile.v0)ARG AMDGPU_VERSION=5.3
2025-05-22 15:52:30 UTC (buildkit.dockerfile.v0)ARG APT_PREF
2025-05-22 15:52:30 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.4.1 AMDGPU_VERSION=6.4.1 APT_PREF=Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600 /bin/sh -c echo "$APT_PREF" > /etc/apt/preferences.d/rocm-pin-600 # buildkit
2025-05-22 15:58:10 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.4.1 AMDGPU_VERSION=6.4.1 APT_PREF=Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates curl libnuma-dev gnupg && curl -sL https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - && printf "deb [arch=amd64] https://repo.radeon.com/rocm/apt/$ROCM_VERSION/ jammy main" | tee /etc/apt/sources.list.d/rocm.list && printf "deb [arch=amd64] https://repo.radeon.com/amdgpu/$AMDGPU_VERSION/ubuntu jammy main" | tee /etc/apt/sources.list.d/amdgpu.list && apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends sudo libelf1 kmod file python3-dev python3-pip rocm-dev rocm-libs build-essential && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit
2025-05-22 15:58:10 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.4.1 AMDGPU_VERSION=6.4.1 APT_PREF=Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600 /bin/sh -c groupadd -g 109 render # buildkit
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ENV PATH=/opt/rocm/llvm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ENV ROCM_PATH=/opt/rocm
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ARG PYTHON_VERSION=3.12
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c mkdir -p /app # buildkit
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)WORKDIR /app
2025-05-28 18:01:09 UTC (buildkit.dockerfile.v0)ENV DEBIAN_FRONTEND=noninteractive
2025-05-28 18:01:49 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c apt-get update -y && apt-get install -y software-properties-common git curl sudo vim less libgfortran5 && for i in 1 2 3; do add-apt-repository -y ppa:deadsnakes/ppa && break || { echo "Attempt $i failed, retrying in 5s..."; sleep 5; }; done && apt-get update -y && apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv python${PYTHON_VERSION}-lib2to3 python-is-python3 && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} && ln -sf /usr/bin/python${PYTHON_VERSION}-config /usr/bin/python3-config && curl -sS https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} && python3 --version && python3 -m pip --version # buildkit
2025-05-28 18:01:53 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' pybind11 Cython # buildkit
2025-05-28 20:24:16 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c dpkg -i /install/*deb && perl -p -i -e 's/, hipblas-common-dev \([^)]*?\), /, /g' /var/lib/dpkg/status && perl -p -i -e 's/, hipblaslt-dev \([^)]*?\), /, /g' /var/lib/dpkg/status && perl -p -i -e 's/, hipblaslt \([^)]*?\), /, /g' /var/lib/dpkg/status # buildkit
2025-05-28 20:24:23 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-05-28 20:24:24 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-05-28 20:24:57 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.4.1-complete
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG HIPBLAS_COMMON_BRANCH=9b80ba8e
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG HIPBLASLT_BRANCH=aa0bda7b
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG LEGACY_HIPBLASLT_OPTION=
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG TRITON_BRANCH=e5be006
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG TRITON_REPO=https://github.com/triton-lang/triton.git
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG PYTORCH_BRANCH=f717b2af
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG PYTORCH_VISION_BRANCH=v0.21.0
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG FA_BRANCH=1a7f4dfa
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG AITER_BRANCH=64876494
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)ARG AITER_REPO=https://github.com/ROCm/aiter.git
2025-05-28 21:23:47 UTC (buildkit.dockerfile.v0)RUN |16 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/dev-ubuntu-22.04:6.4.1-complete HIPBLAS_COMMON_BRANCH=9b80ba8e HIPBLASLT_BRANCH=aa0bda7b LEGACY_HIPBLASLT_OPTION= TRITON_BRANCH=e5be006 TRITON_REPO=https://github.com/triton-lang/triton.git PYTORCH_BRANCH=f717b2af PYTORCH_VISION_BRANCH=v0.21.0 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git FA_BRANCH=1a7f4dfa FA_REPO=https://github.com/Dao-AILab/flash-attention.git AITER_BRANCH=64876494 AITER_REPO=https://github.com/ROCm/aiter.git /bin/sh -c echo "BASE_IMAGE: ${BASE_IMAGE}" > /app/versions.txt && echo "HIPBLAS_COMMON_BRANCH: ${HIPBLAS_COMMON_BRANCH}" >> /app/versions.txt && echo "HIPBLASLT_BRANCH: ${HIPBLASLT_BRANCH}" >> /app/versions.txt && echo "LEGACY_HIPBLASLT_OPTION: ${LEGACY_HIPBLASLT_OPTION}" >> /app/versions.txt && echo "TRITON_BRANCH: ${TRITON_BRANCH}" >> /app/versions.txt && echo "TRITON_REPO: ${TRITON_REPO}" >> /app/versions.txt && echo "PYTORCH_BRANCH: ${PYTORCH_BRANCH}" >> /app/versions.txt && echo "PYTORCH_VISION_BRANCH: ${PYTORCH_VISION_BRANCH}" >> /app/versions.txt && echo "PYTORCH_REPO: ${PYTORCH_REPO}" >> /app/versions.txt && echo "PYTORCH_VISION_REPO: ${PYTORCH_VISION_REPO}" >> /app/versions.txt && echo "FA_BRANCH: ${FA_BRANCH}" >> /app/versions.txt && echo "AITER_BRANCH: ${AITER_BRANCH}" >> /app/versions.txt && echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt # buildkit
2025-05-29 10:42:23 UTC (buildkit.dockerfile.v0)COPY pytorch_afo_testkit/afo/gemm/benchmarks/afo_tune_device_*_full.csv /app # buildkit
2025-05-29 10:42:23 UTC (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_ENABLED=1
2025-05-29 10:42:23 UTC (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_TUNING=0
2025-05-29 10:42:23 UTC (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_FILENAME=/app/afo_tune_device_%d_full.csv
2025-06-05 17:24:46 -0400 (buildkit.dockerfile.v0)ARG ARG_PYTORCH_ROCM_ARCH
2025-06-05 17:24:46 -0400 (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-06-05 17:24:46 -0400 (buildkit.dockerfile.v0)RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c apt-get update -q -y && apt-get install -q -y sqlite3 libsqlite3-dev libfmt-dev libmsgpack-dev libsuitesparse-dev apt-transport-https ca-certificates wget curl # buildkit
2025-06-05 17:24:48 -0400 (buildkit.dockerfile.v0)RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c python3 -m pip install --upgrade pip && pip install setuptools_scm # buildkit
2025-06-05 17:24:50 -0400 (buildkit.dockerfile.v0)RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c apt-get purge -y sccache; python3 -m pip uninstall -y sccache; rm -f "$(which sccache)" # buildkit
2025-06-05 17:24:50 -0400 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-06-05 17:24:50 -0400 (buildkit.dockerfile.v0)WORKDIR /app
2025-06-05 17:24:51 -0400 (buildkit.dockerfile.v0)RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh -c python3 -m pip install --upgrade pip && rm -rf /var/lib/apt/lists/* # buildkit
2025-06-05 17:24:51 -0400 (buildkit.dockerfile.v0)RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh -c case "$(which python3)" in *"/opt/conda/envs/py_3.9"*) rm -rf /opt/conda/envs/py_3.9/lib/python3.9/site-packages/numpy-1.20.3.dist-info/;; *) ;; esac # buildkit
2025-06-05 17:24:55 -0400 (buildkit.dockerfile.v0)RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh -c python3 -m pip install --upgrade huggingface-hub[cli] # buildkit
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh -c cd /install && pip install -U -r requirements/rocm.txt && pip uninstall -y vllm && pip install *.whl # buildkit
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)COPY /benchmarks /app/vllm/benchmarks # buildkit
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)COPY /examples /app/vllm/examples # buildkit
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=1
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)ENV TOKENIZERS_PARALLELISM=false
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)ENV SAFETENSORS_FAST_GPU=1
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)ENV HIP_FORCE_DEV_KERNARG=1
2025-06-05 17:36:12 -0400 (buildkit.dockerfile.v0)CMD ["/bin/bash"]
Delete Tag
Please be careful as this will not just delete the reference but also the actual content!
For example when you have latest and v1.2.3 both pointing to the same image
the deletion of latest will also permanently remove v1.2.3.