Namespace
rocm
Image / Tagvllm:rocm6.3.1_instinct_vllm0.8.3_20250415
Content Digestsha256:ad9062dea3483d59dedb17c67f7c49f30eebd6eb37c3fac0a171fb19696cc845
Details
2025-04-15 14:46:33 -0500
8.97 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
TOKENIZERS_PARALLELISMfalse
Layers
[#000] sha256:7478e0ac0f23f94b2f27848fbcdf804a670fbf8d4bab26df842d40a10cd33059 - 0.32% (29 MB)
[#001] sha256:d96eb0a0e9a06f4e6394ad09df6ae769272b2cdd78d8bc141ed2ecfb5a669b5a - 0.0% (238 Bytes)
[#002] sha256:9feaeb9894ace58074c4b88b97033278ba6dc303478903eab47eaa0441a9b3d3 - 61.45% (5.51 GB)
[#003] sha256:49bac1b84ea2ffb03a73985991c79f8e319eda48465f6affc34d2d25c079672b - 0.0% (656 Bytes)
[#004] sha256:150a2fd90b32bf473f301f5dd70be300d5886c76e36d4c1fa5c94b80b48eefa4 - 0.0% (93 Bytes)
[#005] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#006] sha256:bb2ceb9bfde63da18523731fdab7b21cb2f457c52a978db2092857ff995e9910 - 1.16% (107 MB)
[#007] sha256:c909b68dd0ff4dc2428ef047cf2fb54e817e1a2f90ec6c315a4f0959d4b41502 - 0.72% (66.3 MB)
[#008] sha256:36ace8581d4e935ce38744a4d982fe25f31b23bb3a0f3e0ba25fa7f00a41d1b0 - 9.27% (851 MB)
[#009] sha256:37be2772d609ccf75bb082a728cba0627faa0a95d0f1afc2906104c0c747ba90 - 0.71% (65.2 MB)
[#010] sha256:fafcfe73dd7117f6e67b0803932bf7c55882f1fc9f8bec454fc41863098a63c7 - 2.7% (248 MB)
[#011] sha256:b9bf106e945f52ed0896c42013841e7058aaa0757828b89a85599f1c90018fa2 - 0.02% (1.53 MB)
[#012] sha256:608c98c9178495a433cdc6add7be110289ece463683c2e9fa1cb0c27e037a1a3 - 8.72% (801 MB)
[#013] sha256:9107eb0f6498b4fa974793dfa121bacf203cac2dfe9a89be0806315042a04116 - 2.15% (198 MB)
[#014] sha256:b8ad4ee430d97363c4622f684209269296e7470e028424be149252c3b0896ef5 - 0.0% (425 Bytes)
[#015] sha256:e982a36f30c4033d21c2ea860d189f5ff67e11169f2c66a7f03801398df7cd77 - 0.66% (60.9 MB)
[#016] sha256:4b4a7a4555b081f45f386b7de4d147172061f98df0295b7c159f4eb6cd5f1e0e - 1.07% (98.1 MB)
[#017] sha256:1d8cd6d8bcb3b43066ea4788e19f92523bf1796b7915efd9e702fa8513be4a2a - 0.0% (182 KB)
[#018] sha256:6a9fc10b156e79387a26bad22f6aad6cc3fbbfc9f4caa72bb3205b77aa3360be - 0.0% (215 Bytes)
[#019] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#020] sha256:708e6a3c73c937345e74d59aab2298ed420f7fbaf82856d9e5da275d84a7ca37 - 0.0% (1.97 KB)
[#021] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)
[#022] sha256:e34edacd086c5490a713e80f2cc22a73dce6b493264013771b6cfd7e49a48c45 - 0.06% (5.13 MB)
[#023] sha256:34ed21f92fe26369c7392adb4205573fbf98b61c164a653018aa286bd58c529f - 0.09% (8.07 MB)
[#024] sha256:8c6c1088e08d5004368aff6abf085d9816ec9d8b227ace9c2514ace78e819dea - 10.9% (1000 MB)
[#025] sha256:54d41cda091d65697cd0e5bbc370b30e9501393a02c031ee6beb18fc3c2e415d - 0.0% (116 KB)
[#026] sha256:4aa5b54f063c9677ab576765ed6757b8cd0f81d224bf2d394e0b791cc3609887 - 0.0% (95.4 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-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ENV PATH=/opt/rocm/llvm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ENV ROCM_PATH=/opt/rocm
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ARG PYTHON_VERSION=3.12
2025-04-14 14:25:04 -0500 (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-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)WORKDIR /app
2025-04-14 14:25:04 -0500 (buildkit.dockerfile.v0)ENV DEBIAN_FRONTEND=noninteractive
2025-04-14 14:25:52 -0500 (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 && 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-04-14 14:25:56 -0500 (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 pybind11 Cython # buildkit
2025-04-14 16:30:44 -0500 (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 && 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-04-14 16:30:45 -0500 (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 && 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-04-14 16:30:49 -0500 (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-04-14 16:30:50 -0500 (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-04-14 16:31:08 -0500 (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-04-14 16:31:13 -0500 (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-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3.1-complete
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG HIPBLAS_COMMON_BRANCH=7c1566b
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG HIPBLASLT_BRANCH=0f5d6c6de82bc4461b308ebde3683d54d5a96b30
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG LEGACY_HIPBLASLT_OPTION=
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG RCCL_BRANCH=648a58d
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG RCCL_REPO=https://github.com/ROCm/rccl
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG TRITON_BRANCH=e5be006
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG TRITON_REPO=https://github.com/triton-lang/triton.git
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG PYTORCH_BRANCH=295f2ed4
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG PYTORCH_VISION_BRANCH=v0.21.0
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG PYTORCH_REPO=https://github.com/pytorch/pytorch.git
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG FA_BRANCH=1a7f4dfa
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG AITER_BRANCH=c4a9ce75
2025-04-14 16:31:13 -0500 (buildkit.dockerfile.v0)ARG AITER_REPO=https://github.com/ROCm/aiter.git
2025-04-14 16:31:14 -0500 (buildkit.dockerfile.v0)RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3.1-complete HIPBLAS_COMMON_BRANCH=7c1566b HIPBLASLT_BRANCH=0f5d6c6de82bc4461b308ebde3683d54d5a96b30 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=295f2ed4 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 AITER_BRANCH=c4a9ce75 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 "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-04-15 13:47:43 -0500 (buildkit.dockerfile.v0)COPY pytorch_afo_testkit/afo/gemm/benchmarks/afo_tune_device_*_full.csv /app # buildkit
2025-04-15 13:47:43 -0500 (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_ENABLED=1
2025-04-15 13:47:43 -0500 (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_TUNING=0
2025-04-15 13:47:43 -0500 (buildkit.dockerfile.v0)ENV PYTORCH_TUNABLEOP_FILENAME=/app/afo_tune_device_%d_full.csv
2025-04-15 14:19:11 -0500 (buildkit.dockerfile.v0)ARG ARG_PYTORCH_ROCM_ARCH
2025-04-15 14:19:11 -0500 (buildkit.dockerfile.v0)ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
2025-04-15 14:19:11 -0500 (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-04-15 14:19:13 -0500 (buildkit.dockerfile.v0)RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c python3 -m pip install --upgrade pip && pip install setuptools_scm # buildkit
2025-04-15 14:19:14 -0500 (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-04-15 14:19:14 -0500 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-04-15 14:19:14 -0500 (buildkit.dockerfile.v0)WORKDIR /app
2025-04-15 14:19:15 -0500 (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-04-15 14:19:16 -0500 (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-04-15 14:19:18 -0500 (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-04-15 14:19:18 -0500 (buildkit.dockerfile.v0)ARG BUILD_RPD=1
2025-04-15 14:19:28 -0500 (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-04-15 14:46:33 -0500 (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-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)ARG COMMON_WORKDIR=/app
2025-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)COPY /benchmarks /app/vllm/benchmarks # buildkit
2025-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)COPY /examples /app/vllm/examples # buildkit
2025-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=1
2025-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)ENV TOKENIZERS_PARALLELISM=false
2025-04-15 14:46:33 -0500 (buildkit.dockerfile.v0)ENV HIP_FORCE_DEV_KERNARG=1
2025-04-15 14:46:33 -0500 (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.