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

Manifest digest

sha256:a63dc4b1ee7205173329d2bc4ed87b3583e2d21b8ed91194388614862a7e9621

OS/ARCH

linux/amd64

Compressed size

13.77 GB

Last pushed

4 months by valarlip

Type

Image

Manifest digest

sha256:a63dc4b1ee7205173329d2bc4ed87b3583e2d21b8ed91194388614862a7e9621

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 / 29.18 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.26 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.1116.91 MB

24RUN |11 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1122.14 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.12 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=HEAD186.7 MB

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

37COPY /app/mori /app/mori # buildkit215.15 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 # buildkit825.32 MB

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

44ARG CACHEBUST=17747945630 B

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

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

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

Command

 ARG RELEASE