Sign inSign up

vllm/vllm-omni-rocm:v0.18.0

Manifest digest

sha256:83f39e67e8b5f421b944396252b7ebe00f8365ab2e0a468649bd2b66f1c7aa2a

OS/ARCH

linux/amd64

Compressed size

8.38 GB

Last pushed

6 months by vllmbot

Type

Image

Manifest digest

sha256:83f39e67e8b5f421b944396252b7ebe00f8365ab2e0a468649bd2b66f1c7aa2a

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 29.04 MB
6CMD ["/bin/bash"]0 B
7LABEL [email protected]0 B
8ARG ROCM_VERSION=5.30 B
9ARG AMDGPU_VERSION=5.30 B
10ARG APT_PREF0 B
11RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:237 B
12RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:4.99 GB
13RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:656 B
14ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15ENV ROCM_PATH=/opt/rocm0 B
16ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B
17ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
18ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
19ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
20ENV MORI_GPU_ARCHS=gfx942;gfx9500 B
21ENV HSA_NO_SCRATCH_RECLAIM=10 B
22ARG PYTHON_VERSION=3.120 B
23ENV PYTHON_VERSION=3.120 B
24RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh91 B
25WORKDIR /app32 B
26ENV DEBIAN_FRONTEND=noninteractive0 B
27RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh145.89 MB
28RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh65.81 MB
29RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh4.06 MB
30ARG USE_SCCACHE0 B
31ARG SCCACHE_DOWNLOAD_URL0 B
32ARG SCCACHE_ENDPOINT0 B
33ARG SCCACHE_BUCKET_NAME0 B
34ARG SCCACHE_REGION_NAME0 B
35ARG SCCACHE_S3_NO_CREDENTIALS0 B
36RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=19.11 MB
37RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=1227 B
38ARG USE_SCCACHE0 B
39ENV SCCACHE_BUCKET=vllm-build-sccache0 B
40ENV SCCACHE_REGION=us-west-20 B
41ENV SCCACHE_S3_NO_CREDENTIALS=00 B
42ENV SCCACHE_IDLE_TIMEOUT=00 B
43RUN /bin/sh -c pip install1.68 GB
44ARG BASE_IMAGE0 B
45ARG TRITON_BRANCH0 B
46ARG TRITON_REPO0 B
47ARG PYTORCH_BRANCH0 B
48ARG PYTORCH_VISION_BRANCH0 B
49ARG PYTORCH_REPO0 B
50ARG PYTORCH_VISION_REPO0 B
51ARG PYTORCH_AUDIO_BRANCH0 B
52ARG PYTORCH_AUDIO_REPO0 B
53ARG FA_BRANCH0 B
54ARG FA_REPO0 B
55ARG AITER_BRANCH0 B
56ARG AITER_REPO0 B
57ARG MORI_BRANCH0 B
58ARG MORI_REPO0 B
59RUN |15 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete TRITON_BRANCH=57c693b6 TRITON_REPO=https://github.com/ROCm/triton.git398 B
60ARG ARG_PYTORCH_ROCM_ARCH0 B
61ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
62RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 /bin/sh -c137.38 MB
63RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 /bin/sh -c1.51 KB
64ARG USE_SCCACHE0 B
65RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 USE_SCCACHE=1 /bin/sh32 B
66RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 USE_SCCACHE=1 /bin/sh22.3 MB
67ENV UV_HTTP_TIMEOUT=5000 B
68ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
69ENV UV_LINK_MODE=copy0 B
70ARG USE_SCCACHE0 B
71ARG SCCACHE_DOWNLOAD_URL0 B
72ARG SCCACHE_ENDPOINT0 B
73ARG SCCACHE_BUCKET_NAME0 B
74ARG SCCACHE_REGION_NAME0 B
75ARG SCCACHE_S3_NO_CREDENTIALS0 B
76RUN |8 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 USE_SCCACHE=1 USE_SCCACHE=132 B
77ARG USE_SCCACHE0 B
78ENV SCCACHE_BUCKET=vllm-build-sccache0 B
79ENV SCCACHE_REGION=us-west-20 B
80ENV SCCACHE_S3_NO_CREDENTIALS=00 B
81ENV SCCACHE_IDLE_TIMEOUT=00 B
82ARG COMMON_WORKDIR0 B
83WORKDIR /app32 B
84RUN /bin/sh -c python3 -m1.98 KB
85RUN /bin/sh -c rm -f196 B
86ENV SCCACHE_BUCKET=0 B
87ENV SCCACHE_REGION=0 B
88ENV SCCACHE_S3_NO_CREDENTIALS=0 B
89ENV SCCACHE_IDLE_TIMEOUT=0 B
90RUN /bin/sh -c case "$(which32 B
91RUN /bin/sh -c uv pip6.37 MB
92RUN /bin/sh -c cd /install663.19 MB
93ARG COMMON_WORKDIR0 B
94ARG BASE_IMAGE0 B
95COPY /benchmarks /app/vllm/benchmarks # buildkit215.52 KB
96COPY /examples /app/vllm/examples # buildkit232.46 KB
97COPY /docker /app/vllm/docker # buildkit5.42 KB
98ENV TOKENIZERS_PARALLELISM=false0 B
99ENV SAFETENSORS_FAST_GPU=10 B
100ENV HIP_FORCE_DEV_KERNARG=10 B
101RUN |2 COMMON_WORKDIR=/app BASE_IMAGE=rocm/vllm-dev:base-12919 /bin/sh168 B
102ENV KINETO_CONFIG=/app/libkineto.conf0 B
103RUN |2 COMMON_WORKDIR=/app BASE_IMAGE=rocm/vllm-dev:base-12919 /bin/sh426 B
104CMD ["/bin/bash"]0 B
105ENTRYPOINT ["vllm" "serve"]0 B
106ARG USE_NIGHTLY_BUILD=00 B
107ARG VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab0 B
108ARG ARG_PYTORCH_ROCM_ARCH0 B
109ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
110ARG COMMON_WORKDIR=/app0 B
111WORKDIR /app32 B
112RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=119.37 MB
113RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=32 B
114RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=119 B
115COPY . /app/vllm-omni # buildkit28.23 MB
116RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=517.3 MB
117RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=152 B
118ENTRYPOINT ["vllm" "serve" "--omni"]0 B

Command

 ARG RELEASE