Sign inSign up

amdenterpriseai/aim-radeon-base:2026.11.0

Manifest digest

sha256:b1033cf76cfe63f32fd6d8953810ea2e2d89ad8ed05b5a782d68a1b10ee54b5a

OS/ARCH

linux/amd64

Compressed size

25.33 GB

Last pushed

2 days by amdenterpriseai

Type

Image

Manifest digest

sha256:b1033cf76cfe63f32fd6d8953810ea2e2d89ad8ed05b5a782d68a1b10ee54b5a

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 29.2 MB
5CMD ["/bin/bash"]0 B
6ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
7ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
8ARG AITER_ROCM_ARCH=gfx942;gfx9500 B
9ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
10ARG PREBUILD_KERNELS=10 B
11ARG DEVICE_EXTRAS=device-all0 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=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=193 B
17WORKDIR /app32 B
18ENV DEBIAN_FRONTEND=noninteractive0 B
19RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=1323.27 MB
20RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=119.88 MB
21RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=134.33 MB
22RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=11.64 MB
23ENV PATH=/opt/python/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 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 |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=155.18 MB
29RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=1759.46 KB
30RUN |6 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=16.19 MB
31ARG USE_SCCACHE=00 B
32ARG SCCACHE_DOWNLOAD_URL0 B
33ARG SCCACHE_ENDPOINT=https://minio-frameworks.amd.com0 B
34ARG SCCACHE_BUCKET_NAME=ufb-cache-vllm0 B
35ARG SCCACHE_REGION_NAME=us-east-10 B
36ARG SCCACHE_S3_NO_CREDENTIALS=00 B
37RUN |12 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=132 B
38ARG USE_SCCACHE=00 B
39ARG SCCACHE_S3_KEY_PREFIX=rocm/device-all/0 B
40ENV SCCACHE_BUCKET=ufb-cache-vllm0 B
41ENV SCCACHE_REGION=us-east-10 B
42ENV SCCACHE_S3_NO_CREDENTIALS=00 B
43ENV SCCACHE_IDLE_TIMEOUT=00 B
44ENV SCCACHE_ENDPOINT=https://minio-frameworks.amd.com0 B
45ENV SCCACHE_S3_KEY_PREFIX=rocm/device-all/0 B
46ENV SCCACHE_S3_USE_SSL=on0 B
47ARG PIP_EXTRA_INDEX_URL=https://stable.repo.amd.com/rocm/whl-next0 B
48ENV PIP_EXTRA_INDEX_URL=https://stable.repo.amd.com/rocm/whl-next0 B
49ARG ROCM_SDK_VERSION=10.0.00 B
50ARG TORCH_VERSION=2.12.00 B
51ARG TORCHAUDIO_VERSION=2.11.00 B
52ARG TORCHVISION_VERSION=0.27.00 B
53ARG DEVICE_EXTRAS=device-all0 B
54COPY docker/install_pytorch_wheels.py /tmp/install_pytorch_wheels.py # buildkit5.82 KB
55COPY docker/detect_device_arches.py /tmp/detect_device_arches.py # buildkit3.23 KB
56RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=119.5 GB
57ENV SITE_PACKAGES=/opt/python/lib/python3.14/site-packages0 B
58ENV 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
59ENV 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
60ENV 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
61RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=11.14 MB
62ENV PYTHONPATH=/opt/python/lib/python3.14/site-packages/_rocm_sdk_core/share/amd_smi:0 B
63ENV 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
64ARG USE_SCCACHE=00 B
65RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=132 B
66RUN |18 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=11.6 GB
67ARG BASE_IMAGE=ubuntu:24.040 B
68ARG FA_BRANCH=0e60e3940 B
69ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B
70ARG AITER_BRANCH=v0.1.20.post10 B
71ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
72RUN |23 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 AITER_ROCM_ARCH=gfx942;gfx950 PREBUILD_KERNELS=1262 B
73ARG ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
74ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx12010 B
75ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B
76ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B
77RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 /bin/sh -c150.09 MB
78ARG COMMON_WORKDIR=/app0 B
79WORKDIR /app32 B
80RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh39.96 KB
81RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh32 B
82ENV SCCACHE_BUCKET=0 B
83ENV SCCACHE_REGION=0 B
84ENV SCCACHE_S3_NO_CREDENTIALS=0 B
85ENV SCCACHE_IDLE_TIMEOUT=0 B
86ENV SCCACHE_ENDPOINT=0 B
87ENV SCCACHE_S3_KEY_PREFIX=0 B
88ENV SCCACHE_S3_USE_SSL=0 B
89RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh32 B
90RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh6.49 MB
91RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app /bin/sh777.23 MB
92ARG PYTHON_VERSION=3.140 B
93RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.142.57 MB
94RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14202.89 MB
95ARG COMMON_WORKDIR=/app0 B
96ARG BASE_IMAGE=rocm/ufb-private:vllm-base-ubuntu24.04-py3.14-release-device-all-rocm10.0.0-f46a9dfe2c0 B
97ARG ROCM_VERSION=10.0.00 B
98ENV ROCM_VERSION=10.0.00 B
99COPY /app/vllm /app/vllm # buildkit2.65 GB
100RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14208 B
101ENV TOKENIZERS_PARALLELISM=false0 B
102ENV SAFETENSORS_FAST_GPU=10 B
103ENV HIP_FORCE_DEV_KERNARG=10 B
104RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14166 B
105ENV KINETO_CONFIG=/app/libkineto.conf0 B
106RUN |5 ARG_PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201 COMMON_WORKDIR=/app PYTHON_VERSION=3.14327 B
107CMD ["/bin/bash"]0 B
108ARG PARENT_REGISTRY_PREFIX=docker.io/rocm0 B
109ARG PARENT_REPOSITORY=vllm0 B
110ARG PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.00 B
111ARG INSTALL_LMCACHE=false0 B
112ARG INSTALL_LMCACHE_COMMIT=6fbec460 B
113ENV PYTHONUNBUFFERED=10 B
114ENV DEBIAN_FRONTEND=noninteractive0 B
115ENV PYTHONPATH=/workspace/aim-runtime/src0 B
116ENV HF_HOME=/workspace/model-cache0 B
117ENV HF_DATASETS_CACHE=/workspace/model-cache0 B
118ENV HF_HUB_DISABLE_TELEMETRY=10 B
119ENV AIM_UPSTREAM_IMAGE_REF=docker.io/rocm/vllm:rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.00 B
120ENV AIM_ALLOW_GENERAL_PROFILE_FALLBACK=true0 B
121ENV AIM_ACCELERATOR_FAMILY=radeon0 B
122ENV AIM_ACCELERATOR_TYPE=gpu0 B
123ENV AIM_CONFIG_PATH=/workspace/aim-runtime/config0 B
124RUN |5 PARENT_REGISTRY_PREFIX=docker.io/rocm PARENT_REPOSITORY=vllm PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.032 B
125WORKDIR /workspace99 B
126COPY src/aim_runtime/ /workspace/aim-runtime/src/aim_runtime/ # buildkit201.93 KB
127COPY src/aim_common/ /workspace/aim-runtime/src/aim_common/ # buildkit28.42 KB
128COPY requirements/requirements.txt /workspace/aim-runtime/ # buildkit493 B
129COPY requirements/radeon-requirements.txt /workspace/aim-runtime/ # buildkit730 B
130COPY src/entrypoint.py /workspace/entrypoint.py # buildkit9.23 KB
131RUN |5 PARENT_REGISTRY_PREFIX=docker.io/rocm PARENT_REPOSITORY=vllm PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.032 B
132ARG AIM_BASE_CONFIG_DIR=assets/radeon/base/config0 B
133ARG AIM_BASE_PROFILES_DIR=assets/radeon/base/profiles/general0 B
134COPY assets/radeon/base/profiles/general /workspace/aim-runtime/profiles/general # buildkit741 B
135COPY assets/radeon/base/config /workspace/aim-runtime/config # buildkit567 B
136RUN |7 PARENT_REGISTRY_PREFIX=docker.io/rocm PARENT_REPOSITORY=vllm PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.06.2 MB
137RUN |7 PARENT_REGISTRY_PREFIX=docker.io/rocm PARENT_REPOSITORY=vllm PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.032 B
138RUN |7 PARENT_REGISTRY_PREFIX=docker.io/rocm PARENT_REPOSITORY=vllm PARENT_TAG=rocm10.0.0_ubuntu24.04_py3.14_pytorch_2.12.0_vllm_0.27.0126 B
139COPY LICENSE /workspace/LICENSE # buildkit804 B
140ENTRYPOINT ["./entrypoint.py"]0 B

Command

 ARG RELEASE