Sign inSign up

rocm/vllm-omni:8f2d9e0a9b5fa08028e6006d9bdd991cbf5eef5d

Index digest

sha256:8c9827cd5c01df6d7c5557de50dfa749e0dd2f6db5d25d0a9c5f51dcc48d7517

OS/ARCH

linux/amd64

Compressed size

11.48 GB

Last pushed

about 17 hours by rocmvllm

Type

Image

Manifest digest

sha256:6eedc8c944c3dd714079aef2e405b009115f556439ba04f6571ffa634febf4e2

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=22.040 B
4ADD file ... in / 29.23 MB
5CMD ["/bin/bash"]0 B
7ARG ROCM_VERSION=5.30 B
8ARG AMDGPU_VERSION=5.30 B
9ARG APT_PREF0 B
10RUN |3 ROCM_VERSION=7.2.3 AMDGPU_VERSION=30.30.3 APT_PREF=Package:240 B
11RUN |3 ROCM_VERSION=7.2.3 AMDGPU_VERSION=30.30.3 APT_PREF=Package:6.76 GB
12RUN |3 ROCM_VERSION=7.2.3 AMDGPU_VERSION=30.30.3 APT_PREF=Package:655 B
13ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
14ENV ROCM_PATH=/opt/rocm0 B
15ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B
16ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
17ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
18ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
19ENV HSA_NO_SCRATCH_RECLAIM=10 B
20ARG PYTHON_VERSION=3.120 B
21ENV PYTHON_VERSION=3.120 B
22RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh93 B
23WORKDIR /app32 B
24ENV DEBIAN_FRONTEND=noninteractive0 B
25RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh226.17 MB
26RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh66.23 MB
27RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh4.06 MB
28ARG USE_SCCACHE0 B
29ARG SCCACHE_DOWNLOAD_URL0 B
30ARG SCCACHE_ENDPOINT0 B
31ARG SCCACHE_BUCKET_NAME0 B
32ARG SCCACHE_REGION_NAME0 B
33ARG SCCACHE_S3_NO_CREDENTIALS0 B
34RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=19.11 MB
35RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=1229 B
36ARG USE_SCCACHE0 B
37ENV SCCACHE_BUCKET=vllm-build-sccache0 B
38ENV SCCACHE_REGION=us-west-20 B
39ENV SCCACHE_S3_NO_CREDENTIALS=00 B
40ENV SCCACHE_IDLE_TIMEOUT=00 B
41RUN /bin/sh -c pip install2.71 GB
42COPY /staging/ /opt/rocm/ # buildkit13.43 MB
43ENV ROCPROFILER_QUEUE_INTERPOSITION=00 B
44COPY /staging/ /opt/rocm/ # buildkit14.02 MB
45RUN /bin/sh -c set -eux;14.02 MB
46RUN /bin/sh -c ldconfig #10.49 KB
47ARG BASE_IMAGE0 B
48ARG TRITON_BRANCH0 B
49ARG TRITON_REPO0 B
50ARG PYTORCH_BRANCH0 B
51ARG PYTORCH_VISION_BRANCH0 B
52ARG PYTORCH_REPO0 B
53ARG PYTORCH_VISION_REPO0 B
54ARG PYTORCH_AUDIO_BRANCH0 B
55ARG PYTORCH_AUDIO_REPO0 B
56ARG FA_BRANCH0 B
57ARG FA_REPO0 B
58ARG AITER_BRANCH0 B
59ARG AITER_REPO0 B
60ARG MORI_BRANCH0 B
61ARG MORI_REPO0 B
62ARG ROCM_SYSTEMS_REPO0 B
63ARG ROCPROFILER_SDK_COMMIT0 B
64ARG ROCM_RUNTIME_COMMIT0 B
65RUN |18 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.3-complete TRITON_BRANCH=f0b55c0 TRITON_REPO=https://github.com/ROCm/triton.git500 B
66ARG ARG_PYTORCH_ROCM_ARCH0 B
67ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
68RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c143.15 MB
69RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c92 B
70ARG USE_SCCACHE0 B
71RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/sh32 B
72RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/sh19.38 MB
73ENV UV_HTTP_TIMEOUT=5000 B
74ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
75ENV UV_LINK_MODE=copy0 B
76ENV CCACHE_DIR=/root/.cache/ccache0 B
77ENV CCACHE_COMPILERCHECK=content0 B
78ARG max_jobs0 B
79ENV MAX_JOBS=160 B
80ARG USE_SCCACHE0 B
81ARG SCCACHE_DOWNLOAD_URL0 B
82ARG SCCACHE_ENDPOINT0 B
83ARG SCCACHE_BUCKET_NAME0 B
84ARG SCCACHE_REGION_NAME0 B
85ARG SCCACHE_S3_NO_CREDENTIALS0 B
86RUN |9 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 max_jobs=1632 B
87ARG USE_SCCACHE0 B
88ENV SCCACHE_BUCKET=vllm-build-sccache0 B
89ENV SCCACHE_REGION=us-west-20 B
90ENV SCCACHE_S3_NO_CREDENTIALS=00 B
91ENV SCCACHE_IDLE_TIMEOUT=00 B
92ARG COMMON_WORKDIR0 B
93WORKDIR /app32 B
94ARG NIC_BACKEND0 B
95ARG AINIC_VERSION0 B
96ARG UBUNTU_CODENAME0 B
97RUN |3 NIC_BACKEND=all AINIC_VERSION=1.117.5-a-77 UBUNTU_CODENAME=jammy16.64 MB
98RUN /bin/sh -c python3 -m92 B
99RUN /bin/sh -c rm -f197 B
100ENV SCCACHE_BUCKET=0 B
101ENV SCCACHE_REGION=0 B
102ENV SCCACHE_ENDPOINT=0 B
103ENV SCCACHE_S3_NO_CREDENTIALS=0 B
104ENV SCCACHE_IDLE_TIMEOUT=0 B
105RUN /bin/sh -c case "$(which32 B
106RUN /bin/sh -c uv pip6.53 MB
107RUN /bin/sh -c cd /install754.26 MB
108RUN /bin/sh -c uv pip24.75 MB
109ARG COMMON_WORKDIR0 B
110ARG BASE_IMAGE0 B
111ARG NIC_BACKEND0 B
112ARG AINIC_VERSION0 B
113COPY /benchmarks /app/vllm/benchmarks # buildkit293.98 KB
114COPY /examples /app/vllm/examples # buildkit272.42 KB
115COPY /docker /app/vllm/docker # buildkit41.38 KB
116ENV HSA_ENABLE_IPC_MODE_LEGACY=10 B
117ENV UCX_RMA_PPLN_ENABLE=y0 B
118ENV UCX_ROCM_COPY_SIGPOOL_MAX_ELEMS=inf0 B
119ENV TOKENIZERS_PARALLELISM=false0 B
120ENV SAFETENSORS_FAST_GPU=10 B
121ENV HIP_FORCE_DEV_KERNARG=10 B
122ENV PYTORCH_NVML_BASED_CUDA_CHECK=10 B
123RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:7dcd1182d2cefbba-rocm-base NIC_BACKEND=all168 B
124ENV KINETO_CONFIG=/app/libkineto.conf0 B
125RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:7dcd1182d2cefbba-rocm-base NIC_BACKEND=all595 B
126CMD ["/bin/bash"]0 B
127RUN /bin/sh -c uv pip84.84 MB
128ENTRYPOINT ["vllm" "serve"]0 B
129ARG USE_NIGHTLY_BUILD=00 B
130ARG VLLM_VERSION_OR_COMMIT_HASH=v0.30.00 B
131ARG ARG_PYTORCH_ROCM_ARCH=gfx9420 B
132ENV PYTORCH_ROCM_ARCH=gfx9420 B
133ARG COMMON_WORKDIR=/app0 B
134WORKDIR /app32 B
135RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx942112.47 MB
136RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx94232 B
137RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx9421.33 MB
138COPY tools/install_torchcodec_rocm.sh /tmp/install_torchcodec_rocm.sh # buildkit1.89 KB
139RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx9424.64 MB
140WORKDIR /app/vllm-omni119 B
141RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx942167 B
142COPY pyproject.toml setup.py ./ #8.6 KB
143COPY requirements/ requirements/ # buildkit1.58 KB
144ENV SETUPTOOLS_SCM_PRETEND_VERSION=0.0.00 B
145RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx942482.39 MB
146RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx942150 B
147RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=v0.30.0 ARG_PYTORCH_ROCM_ARCH=gfx942288 B
148ARG COMMON_WORKDIR=/app0 B
149WORKDIR /app/vllm-omni32 B
150COPY . . # buildkit36.06 MB
151CMD ["/bin/bash"]0 B
152ENTRYPOINT []0 B

Command

 ARG RELEASE