Sign inSign up

rocm/infera:atom-v0.2.10

Manifest digest

sha256:e80b113f9f27108318a9d77ac3680dbf1727dca7781174ef5570f82f57a69a37

OS/ARCH

linux/amd64

Compressed size

14.02 GB

Last pushed

9 days by rocm

Type

Image

Manifest digest

sha256:e80b113f9f27108318a9d77ac3680dbf1727dca7781174ef5570f82f57a69a37

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 -c67.54 MB
31ARG ATOM_REPO=https://github.com/ROCm/ATOM.git0 B
32ARG ATOM_COMMIT=main0 B
33RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main183.17 MB
34RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main8.89 MB
35ARG INSTALL_MOONCAKE=10 B
36ARG MOONCAKE_REPO=https://github.com/Jasen2201/Mooncake.git0 B
37ARG MOONCAKE_COMMIT=fix/ionic-mr-and-qp-resource-fixes0 B
38ARG VENV_PYTHON=/opt/venv/bin/python0 B
39RUN |7 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main91.63 MB
40ENV PATH=/usr/local/go/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
41RUN |7 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main214.39 MB
42ARG 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
43RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main499.28 KB
44RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main319.52 MB
45ARG RUST_VERSION=1.94.00 B
46RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main203.15 MB
47ENV 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
48COPY /app/rccl/build/release/*.deb /tmp/rccl/ # buildkit16.97 MB
49RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main18.04 MB
50COPY /app/aiter-test /app/aiter-test # buildkit1.32 GB
51RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main938.32 MB
52RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main92 B
53ARG CACHEBUST=17812548130 B
54RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main618.36 MB
55RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main32 B
56RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main1.94 MB
57RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main15.89 MB
58CMD ["/bin/bash"]0 B
59COPY deploy/docker/patches/atom/ /tmp/atom-patches/ # buildkit4.47 KB
60RUN /bin/sh -c for f20.25 KB
61ARG BUILD_MOONCAKE=10 B
62ARG MOONCAKE_GIT_REF=faae8dd4a6309c3ecd47e0721a83b0250d686fa20 B
63ARG MOONCAKE_HIP_DMABUF=10 B
64COPY deploy/docker/scripts/build_mooncake_rocm.sh /tmp/build_mooncake_rocm.sh # buildkit4.01 KB
65COPY deploy/docker/patches/mooncake_cpp/ /tmp/mooncake_cpp/ # buildkit2.1 KB
66RUN |3 BUILD_MOONCAKE=1 MOONCAKE_GIT_REF=faae8dd4a6309c3ecd47e0721a83b0250d686fa2 MOONCAKE_HIP_DMABUF=1362.44 MB
67WORKDIR /opt/infera116 B
68COPY pyproject.toml README.md ./ #7.13 KB
69COPY infera ./infera # buildkit1.31 MB
70RUN |3 BUILD_MOONCAKE=1 MOONCAKE_GIT_REF=faae8dd4a6309c3ecd47e0721a83b0250d686fa2 MOONCAKE_HIP_DMABUF=123.52 MB
71COPY rust ./rust # buildkit118.41 KB
72RUN |3 BUILD_MOONCAKE=1 MOONCAKE_GIT_REF=faae8dd4a6309c3ecd47e0721a83b0250d686fa2 MOONCAKE_HIP_DMABUF=15.87 MB
73RUN |3 BUILD_MOONCAKE=1 MOONCAKE_GIT_REF=faae8dd4a6309c3ecd47e0721a83b0250d686fa2 MOONCAKE_HIP_DMABUF=1278 B
74COPY deploy/docker/scripts/infera_inject_host_ionic.sh /usr/local/bin/infera-inject-host-ionic # buildkit1.59 KB
75ENTRYPOINT ["/usr/local/bin/infera-inject-host-ionic"]0 B
76CMD ["/bin/bash"]0 B

Command

 ARG RELEASE