rocm/atom-dev:vllm-v0.31.0-nightly_20261021-dsv41-lmcache-6fb1f34c3

Manifest digest

sha256:cfb7d6a0ce67a33dcb41ef8cf5c21f73cb2fa2a0bccdf94eb87e8c5e1d4a95b7

OS/ARCH

linux/amd64

Compressed size

19.95 GB

Last pushed

about 14 hours by valarlip

Type

Image

Manifest digest

sha256:cfb7d6a0ce67a33dcb41ef8cf5c21f73cb2fa2a0bccdf94eb87e8c5e1d4a95b7

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

16

LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0+rocm7.2.4.lw.git3d3aa833

0 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

28ARG BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.00 B

29ENV GPU_ARCH_LIST=gfx942;gfx9500 B

30ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

31ENV ATOM_BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.00 B

32ENV HSA_ENABLE_IPC_MODE_LEGACY=10 B

33RUN |2 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 /bin/sh85.34 MB

34RUN |2 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 /bin/sh32 B

35RUN |2 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 /bin/sh32 B

36RUN |2 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 /bin/sh32 B

37ARG TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/0 B

38ARG TRITON_PIN_VERSION=3.8.0+amd.rocm7.2.0.git7cb7b0590 B

39ARG TRITON_KERNELS_PIN_VERSION=1.0.0+amd.rocm7.2.0.git7cb7b0590 B

40

RUN |5 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

929.68 MB

41ARG ATOM_REPO=https://github.com/ROCm/ATOM.git0 B

42ARG ATOM_COMMIT=6fb1f34c3bccb79dc994b1b8e04353e6b48160130 B

43

RUN |7 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

196.64 MB

44

RUN |7 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

18.11 MB

45ARG INSTALL_MOONCAKE=10 B

46ARG INSTALL_MOONCAKE_STORE=10 B

47ARG MOONCAKE_REPO=https://github.com/kvcache-ai/Mooncake.git0 B

48ARG MOONCAKE_COMMIT=v0.3.14-rc10 B

49ARG USE_HIP_DMABUF=ON0 B

50ARG VENV_PYTHON=/opt/venv/bin/python0 B

51

RUN |13 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

123.83 MB

52ENV PATH=/usr/local/go/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

53

RUN |13 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

204.32 MB

54

ARG 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.deb

0 B

55

RUN |14 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

504.33 KB

56

RUN |14 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

469.68 MB

57ENV MOONCAKE_DISABLE_HIP_DMABUF=10 B

58ARG RUST_VERSION=1.94.00 B

59

RUN |15 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

203.2 MB

60

ENV 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:/bin

0 B

61COPY /rccl-pkgs/ /tmp/rccl/ # buildkit16.97 MB

62

RUN |15 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

18.05 MB

63COPY /app/aiter-test /app/aiter-test # buildkit1.71 GB

64

RUN |15 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

624.57 MB

65

RUN |15 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

92 B

66ARG CACHEBUST=17925963350 B

67ARG ATOM_MESH_FEATURES=0 B

68

RUN |17 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

786.16 MB

69

RUN |17 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

32 B

70

RUN |17 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

2 MB

71

RUN |17 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

14.58 MB

72ARG LMCACHE_WHEEL_SHA256=68c0f6870c82497b4a1fef5c365960119bedfe66715a410457ae45eb6164cc030 B

73ARG LMCACHE_TAG=v0.4.50 B

74

RUN |19 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

315.5 MB

75ARG INSTALL_SA_AIPERF=10 B

76ARG SA_AIPERF_REF=754356e9a39acc6cc6afb242d123bb57c3fb6f750 B

77

RUN |21 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

234.72 MB

78

RUN |21 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

32 B

79ARG TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/0 B

80ARG TRITON_PIN_VERSION=3.8.0+amd.rocm7.2.0.git7cb7b0590 B

81ARG TRITON_KERNELS_PIN_VERSION=1.0.0+amd.rocm7.2.0.git7cb7b0590 B

82

RUN |21 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

32 B

83COPY /staging/ /opt/rocm/ # buildkit14.09 MB

84

RUN |21 GPU_ARCH=gfx942;gfx950 BASE_IMAGE=rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 TRITON_INDEX_URL=https://pypi.amd.com/triton/release_0914/rocm-7.2.0/simple/

14.12 MB

85CMD ["/bin/bash"]0 B

86ARG MAX_JOBS=640 B

87ARG VENV_PYTHON=/opt/venv/bin/python0 B

88ARG VLLM_REPO=https://github.com/vllm-project/vllm.git0 B

89ARG VLLM_COMMIT=db9527a46873454610df6dbedf79a36d6bf1a7f60 B

90ARG INSTALL_LM_EVAL=10 B

91ARG INSTALL_FASTSAFETENSORS=10 B

92LABEL com.rocm.atom.vllm_commit=db9527a46873454610df6dbedf79a36d6bf1a7f60 B

93

ENV PATH=/opt/venv/bin:/root/.cargo/bin:/usr/local/go/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

94ENV MAX_JOBS=640 B

95ENV VLLM_TARGET_DEVICE=rocm0 B

96ENV CMAKE_MAKE_PROGRAM=/usr/local/bin/ninja0 B

97RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git17.05 MB

98RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git614.72 MB

99RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git473.75 MB

100COPY docker/patches/vllm-torch210-compat.patch /tmp/vllm-torch210-compat.patch # buildkit2.06 KB

101RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git361.25 MB

102RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git799.52 MB

103RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git3.44 MB

104RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git1.59 GB

105RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git114.14 MB

106RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git92 B

107RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git473.75 MB

108RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git17.08 MB

109RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git32 B

110CMD ["/bin/bash"]0 B

Command

 ARG RELEASE