Sign inSign up

rocm/vllm:rocm7.13.0_gfx1152_ubuntu24.04_py3.13_pytorch_2.10.0_vllm_0.19.1

Manifest digest

sha256:61b7fa5429d00f8209b36b3923c600df7dedfbfac34bc178d73d17edff3aa793

OS/ARCH

linux/amd64

Compressed size

8.69 GB

Last pushed

about 2 months by rocmdev

Type

Image

Manifest digest

sha256:61b7fa5429d00f8209b36b3923c600df7dedfbfac34bc178d73d17edff3aa793

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.36 MB
5CMD ["/bin/bash"]0 B
6ARG PYTORCH_ROCM_ARCH=gfx11520 B
7ENV PYTORCH_ROCM_ARCH=gfx11520 B
8ARG AITER_ROCM_ARCH=0 B
9ENV AITER_ROCM_ARCH=0 B
10ARG DEVICE_EXTRAS0 B
11ARG PYTHON_VERSION=3.130 B
12ARG MAX_JOBS=640 B
13ENV MAX_JOBS=640 B
14ENV HSA_NO_SCRATCH_RECLAIM=10 B
15RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=93 B
16WORKDIR /app32 B
17ENV DEBIAN_FRONTEND=noninteractive0 B
18RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=327.25 MB
19RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=23.45 MB
20RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=33.29 MB
21RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=1.65 MB
22ENV PATH=/opt/python/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
23ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
24ENV UV_PYTHON=/opt/python/bin/python0 B
25ENV UV_HTTP_TIMEOUT=5000 B
26ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
27ENV UV_LINK_MODE=copy0 B
28RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=55.15 MB
29RUN |5 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=6.18 MB
30ARG USE_SCCACHE=00 B
31ARG SCCACHE_DOWNLOAD_URL0 B
32ARG SCCACHE_ENDPOINT=https://minio-frameworks.amd.com0 B
33ARG SCCACHE_BUCKET_NAME=ufb-cache-vllm0 B
34ARG SCCACHE_REGION_NAME=us-east-10 B
35ARG SCCACHE_S3_NO_CREDENTIALS=00 B
36RUN |11 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=32 B
37ARG USE_SCCACHE=00 B
38ARG SCCACHE_S3_KEY_PREFIX=rocm/gfx1152/0 B
39ARG SCCACHE_S3_ACCESS_KEY=xAwpnZQCYVJo1cQl3u2o0 B
40ARG SCCACHE_S3_SECRET_KEY=0aCxw16qjlRo7QtIEyKClaM8ypcYaLqqZ4SyF2Qz0 B
41ENV SCCACHE_BUCKET=ufb-cache-vllm0 B
42ENV SCCACHE_REGION=us-east-10 B
43ENV SCCACHE_S3_NO_CREDENTIALS=00 B
44ENV SCCACHE_IDLE_TIMEOUT=00 B
45ENV SCCACHE_ENDPOINT=https://minio-frameworks.amd.com0 B
46ENV SCCACHE_S3_KEY_PREFIX=rocm/gfx1152/0 B
47ENV SCCACHE_S3_USE_SSL=on0 B
48ENV AWS_ACCESS_KEY_ID=xAwpnZQCYVJo1cQl3u2o0 B
49ENV AWS_SECRET_ACCESS_KEY=0aCxw16qjlRo7QtIEyKClaM8ypcYaLqqZ4SyF2Qz0 B
50ARG PIP_EXTRA_INDEX_URL=https://rocm.prereleases.amd.com/whl-staging/gfx11520 B
51ENV PIP_EXTRA_INDEX_URL=https://rocm.prereleases.amd.com/whl-staging/gfx11520 B
52ARG ROCM_SDK_VERSION=7.13.0rc20 B
53ARG TORCH_VERSION=2.10.00 B
54ARG TORCHAUDIO_VERSION=2.10.00 B
55ARG TORCHVISION_VERSION=0.25.00 B
56ARG DEVICE_EXTRAS0 B
57RUN |19 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=6.36 GB
58ENV SITE_PACKAGES=/opt/python/lib/python3.13/site-packages0 B
59ENV ROCM_PATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel ROCM_HOME=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel HIP_PATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel HIP_CLANG_PATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel/lib/llvm/bin0 B
60ENV LD_LIBRARY_PATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel/lib:/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib:/opt/python/lib/python3.13/site-packages/_rocm_sdk_core/lib:/opt/python/lib/python3.13/site-packages/_rocm_sdk_core/lib/rocm_sysdeps/lib0 B
61ENV PATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel/bin:/opt/python/lib/python3.13/site-packages/_rocm_sdk_devel/lib/llvm/bin:/opt/python/lib/python3.13/site-packages/_rocm_sdk_core/bin:/opt/python/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
62ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
63RUN |19 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=1.26 MB
64ENV PYTHONPATH=/opt/python/lib/python3.13/site-packages/_rocm_sdk_core/share/amd_smi:0 B
65ARG USE_SCCACHE=00 B
66RUN |19 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=32 B
67RUN |19 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=1.08 MB
68ARG BASE_IMAGE=ubuntu:24.040 B
69ARG FA_BRANCH=0e60e3940 B
70ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B
71ARG AITER_BRANCH=v0.1.10.post20 B
72ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
73RUN |24 PYTORCH_ROCM_ARCH=gfx1152 AITER_ROCM_ARCH= DEVICE_EXTRAS=261 B
74ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
75ARG ARG_PYTORCH_ROCM_ARCH=gfx11520 B
76ENV PYTORCH_ROCM_ARCH=gfx11520 B
77ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B
78ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B
79RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx1152 /bin/sh -c155.64 MB
80ARG COMMON_WORKDIR=/app0 B
81WORKDIR /app32 B
82RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app /bin/sh1.96 MB
83RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app /bin/sh32 B
84ENV SCCACHE_BUCKET=0 B
85ENV SCCACHE_REGION=0 B
86ENV SCCACHE_S3_NO_CREDENTIALS=0 B
87ENV SCCACHE_IDLE_TIMEOUT=0 B
88ENV SCCACHE_ENDPOINT=0 B
89ENV SCCACHE_S3_KEY_PREFIX=0 B
90ENV SCCACHE_S3_USE_SSL=0 B
91ENV AWS_ACCESS_KEY_ID=0 B
92ENV AWS_SECRET_ACCESS_KEY=0 B
93RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app /bin/sh32 B
94RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app /bin/sh7.66 MB
95RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app /bin/sh442.57 MB
96ARG COMMON_WORKDIR=/app0 B
97ARG BASE_IMAGE=rocm/utd-private:vllm-base-ubuntu24.04-py3.13-prereleases-gfx1152-rocm7.13.0rc2-24efb890410 B
98ARG ROCM_VERSION=7.13.0rc20 B
99ENV ROCM_VERSION=7.13.0rc20 B
100COPY /app/vllm /app/vllm # buildkit1.17 GB
101RUN |4 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app BASE_IMAGE=rocm/utd-private:vllm-base-ubuntu24.04-py3.13-prereleases-gfx1152-rocm7.13.0rc2-24efb89041208 B
102ENV TOKENIZERS_PARALLELISM=false0 B
103ENV SAFETENSORS_FAST_GPU=10 B
104ENV HIP_FORCE_DEV_KERNARG=10 B
105RUN |4 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app BASE_IMAGE=rocm/utd-private:vllm-base-ubuntu24.04-py3.13-prereleases-gfx1152-rocm7.13.0rc2-24efb89041168 B
106ENV KINETO_CONFIG=/app/libkineto.conf0 B
107RUN |4 ARG_PYTORCH_ROCM_ARCH=gfx1152 COMMON_WORKDIR=/app BASE_IMAGE=rocm/utd-private:vllm-base-ubuntu24.04-py3.13-prereleases-gfx1152-rocm7.13.0rc2-24efb89041335 B
108CMD ["/bin/bash"]0 B
109/bin/bash0 B
110/bin/bash288 B
111bash -c pip100.86 MB

Command

 ARG RELEASE