Sign inSign up

rocm/vllm-omni:8acdba5770b2287329294e165c013fd545da9f7e

Index digest

sha256:ec3e68d050f41dd9b9836ca4a36ddce6c26065cb636b8fe7e0842599cb26f668

OS/ARCH

linux/amd64

Compressed size

18.41 GB

Last pushed

6 days by rocmvllm

Type

Image

Manifest digest

sha256:e07ad58a85e8a4b4616ee2d2be8cb80d67101faf3bc1d7e9d1e21eb9b11a718e

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=22.040 B
4ADD file ... in / 29.25 MB
5CMD ["/bin/bash"]0 B
6ARG PYTORCH_ROCM_ARCH0 B
7ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
8ARG AITER_ROCM_ARCH0 B
9ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
10ARG SITE_PACKAGES0 B
11ENV SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages0 B
12ENV SDK_DEV=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel SDK_CORE=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_core0 B
13ENV ROCM_PATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel ROCM_HOME=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel HIP_PATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel HIP_CLANG_PATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/lib/llvm/bin0 B
14ENV LD_LIBRARY_PATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/lib:/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib:/usr/local/lib/python3.12/dist-packages/_rocm_sdk_core/lib:/usr/local/lib/python3.12/dist-packages/_rocm_sdk_core/lib/rocm_sysdeps/lib0 B
15ENV PATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/bin:/usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/lib/llvm/bin:/usr/local/lib/python3.12/dist-packages/_rocm_sdk_core/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
16ENV HSA_NO_SCRATCH_RECLAIM=10 B
17ARG PYTHON_VERSION0 B
18ENV PYTHON_VERSION=3.120 B
19RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages93 B
20WORKDIR /app32 B
21ENV DEBIAN_FRONTEND=noninteractive0 B
22RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages278.05 MB
23RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages66.23 MB
24RUN |4 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages3.78 MB
25ARG ROCM_RELEASE_WHEELS_MULTIARCH_URL0 B
26ARG ROCM_NIGHTLY_WHEELS_MULTIARCH_URL0 B
27ARG ROCM_SDK_VERSION0 B
28RUN |7 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages2.16 GB
29ARG TORCH_VERSION0 B
30ARG TORCHAUDIO_VERSION0 B
31ARG TORCHVISION_VERSION0 B
32RUN |10 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages1.28 GB
33SHELL [/bin/bash -c]0 B
34RUN |10 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages10.56 GB
35RUN |10 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages32 B
36RUN |10 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages1.11 MB
37ENV PYTHONPATH=/usr/local/lib/python3.12/dist-packages/_rocm_sdk_core/share/amd_smi0 B
38RUN |10 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages14.28 KB
39ARG USE_SCCACHE0 B
40ARG SCCACHE_DOWNLOAD_URL0 B
41ARG SCCACHE_ENDPOINT0 B
42ARG SCCACHE_BUCKET_NAME0 B
43ARG SCCACHE_REGION_NAME0 B
44ARG SCCACHE_S3_NO_CREDENTIALS0 B
45RUN |16 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages9.11 MB
46RUN |16 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 AITER_ROCM_ARCH=gfx942;gfx950 SITE_PACKAGES=/usr/local/lib/python3.12/dist-packages265 B
47ENV SCCACHE_BUCKET=vllm-build-sccache0 B
48ENV SCCACHE_REGION=us-west-20 B
49ENV SCCACHE_S3_NO_CREDENTIALS=00 B
50ENV SCCACHE_IDLE_TIMEOUT=00 B
51ARG TRITON_VERSION0 B
52ENV ROCPROFILER_QUEUE_INTERPOSITION=00 B
53COPY /staging/lib/ /usr/local/lib/python3.12/dist-packages/_rocm_sdk_devel/lib/ # buildkit15.08 MB
54RUN |1 TRITON_VERSION=3.8.0+git669b31ac /bin/bash -c1.34 MB
55RUN |1 TRITON_VERSION=3.8.0+git669b31ac /bin/bash -c10.39 KB
56RUN |1 TRITON_VERSION=3.8.0+git669b31ac /bin/bash -c560 KB
57RUN |1 TRITON_VERSION=3.8.0+git669b31ac /bin/bash -c1.26 GB
58ARG BASE_IMAGE0 B
59ARG PYTHON_VERSION0 B
60ARG ROCM_SDK_VERSION0 B
61ARG TORCH_VERSION0 B
62ARG TORCHAUDIO_VERSION0 B
63ARG TORCHVISION_VERSION0 B
64ARG PYTORCH_ROCM_ARCH0 B
65ARG AITER_ROCM_ARCH0 B
66ARG TRITON_BRANCH0 B
67ARG TRITON_REPO0 B
68ARG FA_BRANCH0 B
69ARG FA_REPO0 B
70ARG AITER_BRANCH0 B
71ARG AITER_REPO0 B
72ARG MORI_BRANCH0 B
73ARG MORI_REPO0 B
74ARG ROCM_SYSTEMS_REPO0 B
75ARG ROCM_RUNTIME_COMMIT0 B
76ARG TRITON_VERSION0 B
77RUN |20 TRITON_VERSION=3.8.0+git669b31ac BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.12617 B
78ARG ARG_PYTORCH_ROCM_ARCH0 B
79ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
80RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/bash -c142.65 MB
81RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/bash -c92 B
82ARG USE_SCCACHE0 B
83RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/bash32 B
84RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/bash16.48 MB
85ENV UV_HTTP_TIMEOUT=5000 B
86ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
87ENV UV_LINK_MODE=copy0 B
88ENV CCACHE_DIR=/root/.cache/ccache0 B
89ENV CCACHE_COMPILERCHECK=content0 B
90ARG max_jobs0 B
91ENV MAX_JOBS=160 B
92ARG USE_SCCACHE0 B
93ARG SCCACHE_DOWNLOAD_URL0 B
94ARG SCCACHE_ENDPOINT0 B
95ARG SCCACHE_BUCKET_NAME0 B
96ARG SCCACHE_REGION_NAME0 B
97ARG SCCACHE_S3_NO_CREDENTIALS0 B
98RUN |9 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 max_jobs=1632 B
99ARG USE_SCCACHE0 B
100ENV SCCACHE_BUCKET=vllm-build-sccache0 B
101ENV SCCACHE_REGION=us-west-20 B
102ENV SCCACHE_S3_NO_CREDENTIALS=00 B
103ENV SCCACHE_IDLE_TIMEOUT=00 B
104ARG COMMON_WORKDIR0 B
105WORKDIR /app32 B
106ARG NIC_BACKEND0 B
107ARG AINIC_VERSION0 B
108ARG UBUNTU_CODENAME0 B
109RUN |3 NIC_BACKEND=all AINIC_VERSION=1.117.5-a-77 UBUNTU_CODENAME=jammy17.48 MB
110ENV UV_COMPILE_BYTECODE=10 B
111RUN /bin/bash -c python3 -m92 B
112RUN /bin/bash -c rm -f196 B
113ENV SCCACHE_BUCKET=0 B
114ENV SCCACHE_REGION=0 B
115ENV SCCACHE_ENDPOINT=0 B
116ENV SCCACHE_S3_NO_CREDENTIALS=0 B
117ENV SCCACHE_IDLE_TIMEOUT=0 B
118RUN /bin/bash -c case "$(which32 B
119RUN /bin/bash -c uv pip8.46 MB
120RUN /bin/bash -c cd /install884.49 MB
121RUN /bin/bash -c uv pip25.78 MB
122RUN /bin/bash -c EXPECTED=$(grep '^TRITON_SHA256:92 B
123RUN /bin/bash -c EXPECTED=$(grep '^PYBIND11_SHA256:32 B
124ARG COMMON_WORKDIR0 B
125ARG BASE_IMAGE0 B
126ARG NIC_BACKEND0 B
127ARG AINIC_VERSION0 B
128COPY /benchmarks /app/vllm/benchmarks # buildkit303.01 KB
129COPY /examples /app/vllm/examples # buildkit314.95 KB
130COPY /docker /app/vllm/docker # buildkit43.69 KB
131ENV HSA_ENABLE_IPC_MODE_LEGACY=10 B
132ENV UCX_ROCM_COPY_SIGPOOL_MAX_ELEMS=inf0 B
133ENV UCX_ROCM_IPC_SIGPOOL_MAX_ELEMS=inf0 B
134ENV TOKENIZERS_PARALLELISM=false0 B
135ENV SAFETENSORS_FAST_GPU=10 B
136ENV HIP_FORCE_DEV_KERNARG=10 B
137ENV PYTORCH_NVML_BASED_CUDA_CHECK=10 B
138RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:54c0ade8f7b3cc5a-rocm100-base NIC_BACKEND=all168 B
139ENV KINETO_CONFIG=/app/libkineto.conf0 B
140RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:54c0ade8f7b3cc5a-rocm100-base NIC_BACKEND=all709 B
141CMD ["/bin/bash"]0 B
142RUN /bin/bash -c uv pip87.73 MB
143ENTRYPOINT ["vllm" "serve"]0 B
144ARG USE_NIGHTLY_BUILD=10 B
145ARG VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da50 B
146ARG ARG_PYTORCH_ROCM_ARCH=gfx9420 B
147ENV PYTORCH_ROCM_ARCH=gfx9420 B
148ARG COMMON_WORKDIR=/app0 B
149WORKDIR /app32 B
150RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942160.32 MB
151RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942704.15 MB
152RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942216.49 KB
153COPY tools/install_torchcodec_rocm.sh /tmp/install_torchcodec_rocm.sh # buildkit1.9 KB
154RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx9424.21 MB
155WORKDIR /app/vllm-omni119 B
156RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942167 B
157COPY pyproject.toml setup.py ./ #9.44 KB
158COPY requirements/ requirements/ # buildkit1.85 KB
159COPY tools/build_gradio_pillow13.py tools/build_gradio_pillow13.py # buildkit1.7 KB
160COPY tools/gradio_pillow13/ tools/gradio_pillow13/ # buildkit1.95 KB
161ENV VLLM_OMNI_VERSION_OVERRIDE=0.0.00 B
162RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942740 MB
163RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942149 B
164RUN |4 USE_NIGHTLY_BUILD=1 VLLM_VERSION_OR_COMMIT_HASH=4ba787ec9959e566d3b587a796dff07673324da5 ARG_PYTORCH_ROCM_ARCH=gfx942289 B
165ARG COMMON_WORKDIR=/app0 B
166WORKDIR /app/vllm-omni32 B
167COPY . . # buildkit45.13 MB
168CMD ["/bin/bash"]0 B
169ENTRYPOINT []0 B

Command

 ARG RELEASE