Sign inSign up

rocm/atom-dev:mooncake_main

Manifest digest

sha256:7d5be290968998bf5ab76ff91b80dbbee273f3b702533b69e207f17445226e34

OS/ARCH

linux/amd64

Compressed size

14.15 GB

Last pushed

about 15 hours by valarlip

Type

Image

Manifest digest

sha256:7d5be290968998bf5ab76ff91b80dbbee273f3b702533b69e207f17445226e34

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 BASE_IMAGE=ubuntu:24.040 B
7ARG PYTHON_VERSION=3.120 B
8ARG PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa8330 B
9ARG TORCHVISION_VERSION=0.25.0+rocm7.2.4.git82df5f590 B
10ARG TORCHAUDIO_VERSION=2.10.0+rocm7.2.4.git5047768f0 B
11ARG TRITON_VERSION=3.6.0+rocm7.2.4.git4ed888920 B
12ARG APEX_VERSION=1.10.0+rocm7.2.4.git751f5dd50 B
13ARG ROCM_VERSION=7.2.40 B
14ARG UBUNTU_VERSION=24.040 B
15ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.4/0 B
16LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0+rocm7.2.4.lw.git3d3aa8330 B
17ENV DEBIAN_FRONTEND=noninteractive0 B
18RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833258.73 KB
19RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833116.29 MB
20ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/opt/rocm/lib0 B
21RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833114.67 MB
22RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833122.36 MB
23ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
24RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa8337.37 GB
25RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa8331.94 GB
26RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.10.0+rocm7.2.4.lw.git3d3aa833135.12 KB
27ARG GPU_ARCH=gfx942;gfx9500 B
28ENV GPU_ARCH_LIST=gfx942;gfx9500 B
29ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
30RUN |1 GPU_ARCH=gfx942;gfx950 /bin/sh -c77.15 MB
31RUN |1 GPU_ARCH=gfx942;gfx950 /bin/sh -c32 B
32ARG ATOM_REPO=https://github.com/ROCm/ATOM.git0 B
33ARG ATOM_COMMIT=junyyang/reg_dmabuf_mr0 B
34RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr187.15 MB
35RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr13.89 MB
36ARG INSTALL_MOONCAKE=10 B
37ARG MOONCAKE_REPO=https://github.com/kvcache-ai/Mooncake.git0 B
38ARG MOONCAKE_COMMIT=bfca1ce2af8419c50dc8d464820a95d97d43c9300 B
39ARG USE_HIP_DMABUF=ON0 B
40ARG VENV_PYTHON=/opt/venv/bin/python0 B
41RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr117.45 MB
42ENV PATH=/usr/local/go/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
43RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr201.86 MB
44ARG IONIC_DEB_URL=https://repo.radeon.com/amdainic/pensando/ubuntu/1.117.1-a-63/pool/main/r/rdma-core/libionic1_54.0-149.g3304be71_amd64.deb0 B
45RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr503.86 KB
46RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr238.69 MB
47ENV MOONCAKE_DISABLE_HIP_DMABUF=00 B
48ARG RUST_VERSION=1.94.00 B
49RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr203.15 MB
50ENV PATH=/root/.cargo/bin:/usr/local/go/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
51COPY /app/rccl/build/release/*.deb /tmp/rccl/ # buildkit16.97 MB
52RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr18.05 MB
53COPY /app/aiter-test /app/aiter-test # buildkit1.49 GB
54RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr979.68 MB
55RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr92 B
56ARG CACHEBUST=17875609480 B
57RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr701.06 MB
58RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr32 B
59RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr1.95 MB
60RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr16.17 MB
61ARG LMCACHE_TAG=v0.4.50 B
62RUN |12 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr70.36 MB
63ARG INSTALL_SA_AIPERF=10 B
64ARG SA_AIPERF_COMMIT=0d2aa0572ac685943d38c580675c4a61023581d30 B
65RUN |14 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=junyyang/reg_dmabuf_mr209.85 MB
66CMD ["/bin/bash"]0 B

Command

 ARG RELEASE