Sign inSign up

rocm/vllm:rocm7.14.1_rdna_ubuntu24.04_py3.14_pytorch_2.11_vllm_0.23.0

Manifest digest

sha256:19ad8dc5fb3012f2d5810995f73e8bf069302056d2b4aa6fdf2d2ae5fa9a68ab

OS/ARCH

linux/amd64

Compressed size

23.23 GB

Last pushed

about 1 month by rocm

Type

Image

Manifest digest

sha256:19ad8dc5fb3012f2d5810995f73e8bf069302056d2b4aa6fdf2d2ae5fa9a68ab

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.37 MB
5CMD ["/bin/bash"]0 B
6ARG PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
7ENV PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
8ARG AITER_ROCM_ARCH=0 B
9ENV AITER_ROCM_ARCH=0 B
10ARG PREBUILD_KERNELS=00 B
11ARG DEVICE_EXTRAS=device-all-rdna0 B
12ARG PYTHON_VERSION=3.140 B
13ARG MAX_JOBS=640 B
14ENV MAX_JOBS=640 B
15ENV HSA_NO_SCRATCH_RECLAIM=10 B
16RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=093 B
17WORKDIR /app32 B
18ENV DEBIAN_FRONTEND=noninteractive0 B
19RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=0327.1 MB
20RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=019.06 MB
21RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=034.33 MB
22RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=01.64 MB
23ENV PATH=/opt/python/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
24ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
25ENV UV_PYTHON=/opt/python/bin/python0 B
26ENV UV_HTTP_TIMEOUT=5000 B
27ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
28ENV UV_LINK_MODE=copy0 B
29RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=055.2 MB
30RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=0759.37 KB
31RUN |6 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=06.19 MB
32ARG USE_SCCACHE=00 B
33ARG SCCACHE_DOWNLOAD_URL0 B
34ARG SCCACHE_ENDPOINT=https://minio-frameworks.amd.com0 B
35ARG SCCACHE_BUCKET_NAME=ufb-cache-vllm0 B
36ARG SCCACHE_REGION_NAME=us-east-10 B
37ARG SCCACHE_S3_NO_CREDENTIALS=00 B
38RUN |12 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=032 B
39ARG USE_SCCACHE=00 B
40ARG SCCACHE_S3_KEY_PREFIX=rocm/device-all-rdna/0 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/device-all-rdna/0 B
47ENV SCCACHE_S3_USE_SSL=on0 B
48ARG PIP_EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch0 B
49ENV PIP_EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch0 B
50ARG ROCM_SDK_VERSION=7.14.10 B
51ARG TORCH_VERSION=2.11.00 B
52ARG TORCHAUDIO_VERSION=2.11.00 B
53ARG TORCHVISION_VERSION=0.26.00 B
54ARG DEVICE_EXTRAS=device-all-rdna0 B
55COPY docker/install_pytorch_wheels.py /tmp/install_pytorch_wheels.py # buildkit5.82 KB
56COPY docker/detect_device_arches.py /tmp/detect_device_arches.py # buildkit3.23 KB
57COPY docker/ensure-flydsl-rocm-toolkit-layout.py /tmp/ensure-flydsl-rocm-toolkit-layout.py # buildkit2.12 KB
58RUN |18 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=018.94 GB
59ENV SITE_PACKAGES=/opt/python/lib/python3.14/site-packages0 B
60ENV ROCM_PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel ROCM_HOME=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel HIP_PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel HIP_CLANG_PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/lib/llvm/bin0 B
61ENV LD_LIBRARY_PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/lib:/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib:/opt/python/lib/python3.14/site-packages/_rocm_sdk_core/lib:/opt/python/lib/python3.14/site-packages/_rocm_sdk_core/lib/rocm_sysdeps/lib0 B
62ENV PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/bin:/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/lib/llvm/bin:/opt/python/lib/python3.14/site-packages/_rocm_sdk_core/bin:/opt/python/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
63ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
64RUN |18 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=01.14 MB
65ENV PYTHONPATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_core/share/amd_smi:0 B
66ENV CMAKE_PREFIX_PATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel/lib/cmake: CMAKE_HIP_COMPILER_ROCM_ROOT=/opt/python/lib/python3.14/site-packages/_rocm_sdk_devel0 B
67ARG USE_SCCACHE=00 B
68RUN |18 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=032 B
69RUN |18 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=01.04 MB
70ARG BASE_IMAGE=ubuntu:24.040 B
71ARG FA_BRANCH=0e60e3940 B
72ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B
73ARG AITER_BRANCH=release/v0.1.13.post10 B
74ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
75RUN |23 PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH= PREBUILD_KERNELS=0267 B
76ENV FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE0 B
77ARG ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
78ENV PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
79ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B
80ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B
81RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 /bin/sh -c150.12 MB
82ARG COMMON_WORKDIR=/app0 B
83WORKDIR /app32 B
84RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh39.96 KB
85RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh32 B
86ENV SCCACHE_BUCKET=0 B
87ENV SCCACHE_REGION=0 B
88ENV SCCACHE_S3_NO_CREDENTIALS=0 B
89ENV SCCACHE_IDLE_TIMEOUT=0 B
90ENV SCCACHE_ENDPOINT=0 B
91ENV SCCACHE_S3_KEY_PREFIX=0 B
92ENV SCCACHE_S3_USE_SSL=0 B
93RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh32 B
94RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh6.7 MB
95RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh856.78 MB
96ARG PYTHON_VERSION=3.140 B
97RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.142.58 MB
98RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14202.89 MB
99ARG COMMON_WORKDIR=/app0 B
100ARG BASE_IMAGE=rocm/ufb-private:vllm-base-ubuntu24.04-py3.14-release-device-all-rdna-rocm7.14.1-9ddef711790 B
101ARG ROCM_VERSION=7.14.10 B
102ENV ROCM_VERSION=7.14.10 B
103COPY /app/vllm /app/vllm # buildkit2.64 GB
104RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14208 B
105ENV TOKENIZERS_PARALLELISM=false0 B
106ENV SAFETENSORS_FAST_GPU=10 B
107ENV HIP_FORCE_DEV_KERNARG=10 B
108RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14168 B
109ENV KINETO_CONFIG=/app/libkineto.conf0 B
110RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14339 B
111CMD ["/bin/bash"]0 B

Command

 ARG RELEASE