rocm/atom:rocm7.2.0-ubuntu24.04-pytorch2.9-atom0.1.1

Manifest digest

sha256:b708e207930067426c5b0a2eb5e3b5df0ae15f26bbf473ac9188b86c9e14c734

OS/ARCH

linux/amd64

Compressed size

17.32 GB

Last pushed

5 months by valarlip

Type

Image

Manifest digest

sha256:b708e207930067426c5b0a2eb5e3b5df0ae15f26bbf473ac9188b86c9e14c734

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 ROCM_VERSION=7.20 B

11ARG AMDGPU_VERSION=7.2.70200-10 B

12

ARG TORCH_WHEEL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.2/43/torch-2.9.1+rocm7.2.0.lw.git7e1940d4-cp312-cp312-linux_x86_64.whl

0 B

13

ARG APEX_WHEEL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.2/43/apex-1.9.0+rocm7.2.0.gite37ed124-cp312-cp312-linux_x86_64.whl

0 B

14

ARG TORCHAUDIO_WHEEL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.2/43/torchaudio-2.9.0+rocm7.2.0.gite3c6ee2b-cp312-cp312-linux_x86_64.whl

0 B

15

ARG TORCHVISION_WHEEL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.2/43/torchvision-0.24.0+rocm7.2.0.gitb919bd0c-cp312-cp312-linux_x86_64.whl

0 B

16

ARG TRITON_WHEEL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.2/43/triton-3.5.1+rocm7.2.0.gita272dfa8-cp312-cp312-linux_x86_64.whl

0 B

17ENV DEBIAN_FRONTEND=noninteractive0 B

18RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1258.73 KB

19RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1112.24 MB

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

21ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B

22RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1115.68 MB

23RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1121.49 MB

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

25RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.116.67 KB

26RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1448.48 KB

27RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.17.74 GB

28RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.12.01 GB

29RUN |10 BASE_IMAGE=ubuntu:24.04 PYTHON_VERSION=3.12 PYTORCH_VERSION=2.9.1117.41 KB

30ARG GPU_ARCH=gfx942;gfx9500 B

31ENV GPU_ARCH_LIST=gfx942;gfx9500 B

32ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

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

34ARG ATOM_COMMIT=HEAD0 B

35ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B

36ARG AITER_COMMIT=HEAD0 B

37ARG MORI_COMMIT=b0dce4beebeb1f26c784eee17d5fd9785ee9447f0 B

38ARG PREBUILD_KERNELS=10 B

39ARG MAX_JOBS=640 B

40RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD5.18 MB

41RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD185.11 MB

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

43RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD17.67 MB

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

45RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD265.54 MB

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

47ARG RCCL_REPO=https://github.com/ROCm/rccl.git0 B

48ARG RCCL_BRANCH=29e1567b95e28823b0beb1a988adc587bfab5b4f0 B

49ARG RCCL_SRC_DIR=/app/rccl/0 B

50RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD92 B

51RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD420.98 MB

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

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

54RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD341 B

55RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD4.99 GB

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

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

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

59RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD1.19 GB

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

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

62RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD55.84 MB

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

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

Command

 ARG RELEASE