Namespace
rocm
Image / Tagvllm:rocm6.3.1_instinct_vllm0.7.3_20250311
Content Digestsha256:de0a2649b735f45b7ecab8813eb7b19778ae1f40591ca1196b07bc29c42ed4a3
Details
2025-03-05 10:24:56 -0600
8.29 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
RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES1
ROCM_PATH/opt/rocm
TOKENIZERS_PARALLELISMfalse
Layers
[#000] sha256:6414378b647780fee8fd903ddb9541d134a1947ce092d08bdeb23a54cb3684ac - 0.33% (28.2 MB)
[#001] sha256:d96eb0a0e9a06f4e6394ad09df6ae769272b2cdd78d8bc141ed2ecfb5a669b5a - 0.0% (238 Bytes)
[#002] sha256:9feaeb9894ace58074c4b88b97033278ba6dc303478903eab47eaa0441a9b3d3 - 66.46% (5.51 GB)
[#003] sha256:49bac1b84ea2ffb03a73985991c79f8e319eda48465f6affc34d2d25c079672b - 0.0% (656 Bytes)
[#004] sha256:7578fc054eba61bf4a2b3821b59ae4a6e80c87a1cb45eb52042276f7a0ac3f0b - 0.0% (93 Bytes)
[#005] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#006] sha256:d3d2c23da97158e1ccf22387412fad32ab915c3c907d28a8e9570fa975687af8 - 1.23% (104 MB)
[#007] sha256:3ca9855b5ba4a86f2336f73c49319f862877067f06e909bcc4b75ec2c45468c2 - 0.78% (66.2 MB)
[#008] sha256:a1ead04827e8c337b08c533af7c74d564af367e3903dd51a597af638f00a6b57 - 7.44% (632 MB)
[#009] sha256:646c58ca92024aa912388471fdf2f2d820ac70701aa48c4d2b295c2d1285c85e - 0.2% (17.4 MB)
[#010] sha256:43623189bd0312ef8543a9f0f60b7503121ec68186d5403028ab58f4d71a1e03 - 2.93% (248 MB)
[#011] sha256:076e57fdceeb3c75a355234db77945cfee08c35deaa73f6b8fc5ed53858c0643 - 0.02% (1.53 MB)
[#012] sha256:e4fbcb87720a4af2ec33967888d773d69c75281b4c333f56dd83c21d2a070d0b - 8.23% (699 MB)
[#013] sha256:323ca9f7737874e8ff4b4b398c7f6470f132b6f0bd989e3c5fc7aedf920d27c7 - 0.48% (40.7 MB)
[#014] sha256:a2db6d9378616be993a3de05989600d070a0d1ad3ffdc2fefb53175a4aa5e0b4 - 1.85% (157 MB)
[#015] sha256:cfc4546409c0448bb9974756393d213e5aaad2d7e51e1d84667a788dd400e805 - 1.08% (91.7 MB)
[#016] sha256:d1dc04ad70c9fc5b6ec9907cbd6494c9f68460d5d1566b378036afc1af81db27 - 0.0% (402 Bytes)
[#017] sha256:a2537e39d1a638c557c7396ffbaf244c88c2a032aa5306e5da292efd92b817ef - 1.25% (106 MB)
[#018] sha256:9472703d12dc2a41edaf260e465d401a6dd3b9f0a3abc924019e33296daf5290 - 0.0% (182 KB)
[#019] sha256:87e598fb4b8d12006ecb74b83bcb5f0d46493777447640cd14a02be1d61dccb6 - 0.0% (217 Bytes)
[#020] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#021] sha256:0b6b1da4061cb0006b464c6eb15484a1cb22940062096c1ced25cc3cf7ddbc65 - 0.0% (1.98 KB)
[#022] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#023] sha256:32689f9df4ff267224f91f8902e2dcafb2c3f1e175c8f4d74909f5a7214679bb - 0.06% (5.08 MB)
[#024] sha256:003716a47444244cdb61aadc18dd709b3d706b6087f46c84d144f6be46417ce7 - 0.09% (8.01 MB)
[#025] sha256:eab10a82eea522587abd93ad752b13faa85d6efd831d1d962f50a7417479879d - 7.55% (641 MB)
[#026] sha256:62a9a7613bc83fa05ca25312582d83226f60ffa74ef47d9084b738aac40bc10f - 0.0% (101 KB)
[#027] sha256:10a1918a90e35da4b6e20009f906825e0c8f079acd8c215488aac9623616f58c - 0.0% (82 KB)
History
/bin/sh -c #(nop) ARG RELEASE
2024-09-11 16:25:16 UTC/bin/sh -c #(nop) ARG LAUNCHPAD_BUILD_ARCH
2024-09-11 16:25:16 UTC/bin/sh -c #(nop) LABEL org.opencontainers.image.ref.name=ubuntu
2024-09-11 16:25:16 UTC/bin/sh -c #(nop) LABEL org.opencontainers.image.version=22.04
2024-09-11 16:25:17 UTC/bin/sh -c #(nop) ADD file:ebe009f86035c175ba244badd298a2582914415cf62783d510eab3a311a5d4e1 in /
2024-09-11 16:25:18 UTC/bin/sh -c #(nop) CMD ["/bin/bash"]
2024-12-21 00:57:30 UTC (buildkit.dockerfile.v0)LABEL maintainer=dl.mlsedevops@amd.com
2024-12-21 00:57:30 UTC (buildkit.dockerfile.v0)ARG ROCM_VERSION=5.3
2024-12-21 00:57:30 UTC (buildkit.dockerfile.v0)ARG AMDGPU_VERSION=5.3
2024-12-21 00:57:30 UTC (buildkit.dockerfile.v0)ARG APT_PREF
2024-12-21 00:57:30 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.3.1 AMDGPU_VERSION=6.3.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
2024-12-21 01:03:24 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.3.1 AMDGPU_VERSION=6.3.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
2024-12-21 01:03:24 UTC (buildkit.dockerfile.v0)RUN |3 ROCM_VERSION=6.3.1 AMDGPU_VERSION=6.3.1 APT_PREF=Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600 /bin/sh -c groupadd -g 109 render # buildkit
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ENV PATH=/opt/rocm/llvm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ENV ROCM_PATH=/opt/rocm
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ARG PYTHON_VERSION=3.12
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c mkdir -p /app # buildkit
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)WORKDIR /app
2025-02-22 21:53:24 UTC (buildkit.dockerfile.v0)ENV DEBIAN_FRONTEND=noninteractive
2025-02-22 21:54:03 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c apt-get update -y && apt-get install -y software-properties-common git curl sudo vim less && add-apt-repository ppa:deadsnakes/ppa && 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-02-22 21:54:08 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c pip install -U packaging cmake ninja wheel setuptools pybind11 Cython # buildkit
2025-02-24 21:11:39 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c dpkg -i /install/*deb && sed -i 's/, hipblaslt-dev \(.*\), hipcub-dev/, hipcub-dev/g' /var/lib/dpkg/status && sed -i 's/, hipblaslt \(.*\), hipfft/, hipfft/g' /var/lib/dpkg/status # buildkit
2025-02-24 21:11:40 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c dpkg -i /install/*deb && sed -i 's/, rccl-dev \(.*\), rocalution/, rocalution/g' /var/lib/dpkg/status && sed -i 's/, rccl \(.*\), rocalution/, rocalution/g' /var/lib/dpkg/status # buildkit
2025-02-24 21:11:44 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-02-24 21:11:45 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-02-24 21:12:03 UTC (buildkit.dockerfile.v0)RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 /bin/sh -c pip install /install/*.whl # buildkit
2025-02-24 21:12:03 UTC (buildkit.dockerfile.v0)ARG AITER_BRANCH=41297e56
2025-02-24 21:12:03 UTC (buildkit.dockerfile.v0)ARG AITER_REPO=https://github.com/ROCm/aiter.git
2025-02-28 00:11:03 UTC (buildkit.dockerfile.v0)RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 AITER_BRANCH=41297e56 AITER_REPO=https://github.com/ROCm/aiter.git /bin/sh -c git clone --recursive ${AITER_REPO} # buildkit
2025-02-28 00:17:06 UTC (buildkit.dockerfile.v0)RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 AITER_BRANCH=41297e56 AITER_REPO=https://github.com/ROCm/aiter.git /bin/sh -c cd aiter && git checkout ${AITER_BRANCH} && git submodule update --init --recursive && pip install -r requirements.txt # buildkit
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 AITER_BRANCH=41297e56 AITER_REPO=https://github.com/ROCm/aiter.git /bin/sh -c pip install pyyaml && cd aiter && PREBUILD_KERNELS=1 GPU_ARCHS=gfx942 python3 setup.py develop && pip show aiter # buildkit
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3.1-complete
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG HIPBLAS_COMMON_BRANCH=7c1566b
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG HIPBLASLT_BRANCH=db8e93b4
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG LEGACY_HIPBLASLT_OPTION=
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG RCCL_BRANCH=648a58d
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG RCCL_REPO=https://github.com/ROCm/rccl
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG TRITON_BRANCH=e5be006
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG TRITON_REPO=https://github.com/triton-lang/triton.git
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG PYTORCH_BRANCH=6c0e7463
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG PYTORCH_VISION_BRANCH=v0.21.0
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG PYTORCH_REPO=https://github.com/pytorch/pytorch.git
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG FA_BRANCH=1a7f4dfa
2025-02-28 00:26:43 UTC (buildkit.dockerfile.v0)ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git
2025-02-28 00:26:44 UTC (buildkit.dockerfile.v0)RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942 PYTHON_VERSION=3.12 AITER_BRANCH=41297e56 AITER_REPO=https://github.com/ROCm/aiter.git BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3.1-complete HIPBLAS_COMMON_BRANCH=7c1566b HIPBLASLT_BRANCH=db8e93b4 LEGACY_HIPBLASLT_OPTION= RCCL_BRANCH=648a58d RCCL_REPO=https://github.com/ROCm/rccl TRITON_BRANCH=e5be006 TRITON_REPO=https://github.com/triton-lang/triton.git PYTORCH_BRANCH=6c0e7463 PYTORCH_VISION_BRANCH=v0.21.0 PYTORCH_REPO=https://github.com/pytorch/pytorch.git PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git FA_BRANCH=1a7f4dfa FA_REPO=https://github.com/Dao-AILab/flash-attention.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 "RCCL_BRANCH: ${RCCL_BRANCH}" >> /app/versions.txt && echo "RCCL_REPO: ${RCCL_REPO}" >> /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-02-28 15:07:56 -0600 (buildkit.dockerfile.v0)ARG ARG_PYTORCH_ROCM_ARCH
2025-02-28 15:07:56 -0600 (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942
2025-02-28 15:07:56 -0600 (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 # buildkit
2025-02-28 15:07:58 -0600 (buildkit.dockerfile.v0)RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c python3 -m pip install --upgrade pip && pip install setuptools_scm # buildkit
2025-02-28 15:07:59 -0600 (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-02-28 15:07:59 -0600 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-02-28 15:07:59 -0600 (buildkit.dockerfile.v0)WORKDIR /app
2025-02-28 15:08:00 -0600 (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-02-28 15:08:01 -0600 (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-02-28 15:08:03 -0600 (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-02-28 15:08:03 -0600 (buildkit.dockerfile.v0)ARG BUILD_RPD=1
2025-02-28 15:08:13 -0600 (buildkit.dockerfile.v0)RUN |3 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app BUILD_RPD=1 /bin/sh -c if [ ${BUILD_RPD} -eq "1" ]; then git clone -b nvtx_enabled https://github.com/ROCm/rocmProfileData.git && cd rocmProfileData/rpd_tracer && pip install -r requirements.txt && cd ../ && make && make install && cd hipMarker && python3 setup.py install ; fi # buildkit
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)RUN |3 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app BUILD_RPD=1 /bin/sh -c cd /install && pip install -U -r requirements-rocm.txt && pip uninstall -y vllm && pip install *.whl # buildkit
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)COPY /benchmarks /app/vllm/benchmarks # buildkit
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)COPY /examples /app/vllm/examples # buildkit
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=1
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)ENV TOKENIZERS_PARALLELISM=false
2025-03-05 10:24:56 -0600 (buildkit.dockerfile.v0)ENV HIP_FORCE_DEV_KERNARG=1
2025-03-05 10:24:56 -0600 (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.