rocm/atom:rocm7.2.1-ubuntu24.04-pytorch2.9.1-atom0.1.2-vllm0.17.0

Manifest digest

sha256:f5633ac8fcb07a526fb717f1f06310a7fc1c81d0386832a0763d37c95f1af1c6

OS/ARCH

linux/amd64

Compressed size

17.88 GB

Last pushed

3 months by valarlip

Type

Image

Manifest digest

sha256:f5633ac8fcb07a526fb717f1f06310a7fc1c81d0386832a0763d37c95f1af1c6

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.ref.name=ubuntu0 B

4LABEL org.opencontainers.image.version=24.040 B

5ADD file ... in / 28.35 MB

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

7ARG BASE_IMAGE=ubuntu:24.040 B

8ARG PYTHON_VERSION=3.120 B

9ARG PYTORCH_VERSION=2.9.10 B

10ARG TORCHVISION_VERSION=0.24.00 B

11ARG TORCHAUDIO_VERSION=2.9.00 B

12ARG TRITON_VERSION=3.5.10 B

13ARG APEX_VERSION=1.9.00 B

14ARG ROCM_VERSION=7.2.10 B

15ARG AMDGPU_VERSION=7.2.1.702010 B

16ARG UBUNTU_VERSION=24.040 B

17ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/0 B

18LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.9.10 B

19ENV DEBIAN_FRONTEND=noninteractive0 B

20RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1258.73 KB

21RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1116.27 MB

22ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/opt/rocm/lib0 B

23RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1115.85 MB

24RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1122.2 MB

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

26RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.17.63 GB

27RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.11.93 GB

28RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1135.1 KB

29ARG GPU_ARCH=gfx942;gfx9500 B

30ENV GPU_ARCH_LIST=gfx942;gfx9500 B

31ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

32RUN |1 GPU_ARCH=gfx942;gfx950 /bin/sh -c54.18 MB

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

34ARG ATOM_COMMIT=HEAD0 B

35RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD187.48 MB

36COPY /opt/venv /opt/venv # buildkit2.03 GB

37COPY /app/mori /app/mori # buildkit216.25 MB

38RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

39COPY /app/rccl/build/release/*.deb /tmp/rccl/ # buildkit16.97 MB

40RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD18.04 MB

41RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD312.38 MB

42COPY /app/aiter-test /app/aiter-test # buildkit906.11 MB

43RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD145.1 MB

44ARG CACHEBUST=17751403100 B

45RUN |4 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD66.36 MB

46RUN |4 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

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

48ARG MAX_JOBS=640 B

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

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

51ARG VLLM_COMMIT=b31e9326a7d9394aab8c767f8ebe225c65594b600 B

52ARG INSTALL_LM_EVAL=10 B

53ARG INSTALL_FASTSAFETENSORS=10 B

54ARG ROCM_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/0 B

55ARG ROCM_TORCHVISION_VERSION=0.24.0+rocm7.2.1.gitb919bd0c0 B

56ARG ROCM_TORCHAUDIO_VERSION=2.9.0+rocm7.2.1.gite3c6ee2b0 B

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

58ENV MAX_JOBS=640 B

59ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/torch/lib:/opt/rocm/lib0 B

60ENV VLLM_TARGET_DEVICE=rocm0 B

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

62RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git558.07 MB

63RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git424.39 MB

64RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git212.95 MB

65RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git957.28 MB

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

67RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git917.1 MB

68RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git79.1 MB

69RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git559.76 MB

70RUN |9 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git424.4 MB

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

Command

 ARG RELEASE