rocm/atom-dev:ROCm_10.1.0

Manifest digest

sha256:c9f5678138ba5ec44c1b35edb3f92568bd3d274d91c5079c720348335d901597

OS/ARCH

linux/amd64

Compressed size

15.09 GB

Last pushed

1 day by valarlip

Type

Image

Manifest digest

sha256:c9f5678138ba5ec44c1b35edb3f92568bd3d274d91c5079c720348335d901597

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

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

4ADD file ... in / 28.39 MB

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

6ARG GPU_ARCH=gfx942;gfx9500 B

7ARG ROCM_SDK_VERSION=10.1.00 B

8ARG ROCM_TORCH_VERSION=2.13.00 B

9ARG ROCM_TORCHVISION_VERSION=0.28.00 B

10ARG ROCM_TORCHAUDIO_VERSION=2.11.0.30 B

11ARG ROCM_TRITON_VERSION=3.8.0+git669b31ac0 B

12ARG ROCM_INDEX_URL=https://stable.repo.amd.com/rocm/whl-next/0 B

13ENV ROCM_INDEX_URL=https://stable.repo.amd.com/rocm/whl-next/0 B

14RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0151.04 MB

15ENV VIRTUAL_ENV=/opt/venv0 B

16RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.04 MB

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

18RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.06.31 MB

19RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.04.68 GB

20RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.03.67 GB

21RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

22ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

23ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

24ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

25ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

26ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

27RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.01.22 KB

28RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0167 B

29RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0166 B

30RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

31ENV PIP_CONSTRAINT=/etc/atom/constraints/torch-rocm.txt0 B

32RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0222 B

33ARG GPU_ARCH=gfx942;gfx9500 B

34ARG BASE_IMAGE=rocm10-base0 B

35ENV GPU_ARCH_LIST=gfx942;gfx9500 B

36ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

37ENV ATOM_BASE_IMAGE=rocm10-base0 B

38ENV HSA_ENABLE_IPC_MODE_LEGACY=10 B

39RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0149.95 MB

40RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.09.41 KB

41RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0966.08 KB

42RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

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

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

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

46RUN |11 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

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

48ARG ATOM_COMMIT=junyyang/rocm-10.1-hip7.160 B

49RUN |13 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0309.5 MB

50RUN |13 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.07.54 MB

51ARG INSTALL_MOONCAKE=10 B

52ARG INSTALL_MOONCAKE_STORE=10 B

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

54ARG MOONCAKE_COMMIT=v0.3.14-rc10 B

55ARG USE_HIP_DMABUF=ON0 B

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

57RUN |19 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0129.22 MB

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

59RUN |19 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0213.65 MB

60

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

61RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0304.4 KB

62RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0469.68 MB

63ENV MOONCAKE_DISABLE_HIP_DMABUF=10 B

64ARG RUST_VERSION=1.94.00 B

65RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0203.2 MB

66ENV PATH=/root/.cargo/bin:/usr/local/go/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

67COPY /rccl-pkgs/ /tmp/rccl/ # buildkit120 B

68RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0126 B

69COPY /app/aiter-test /app/aiter-test # buildkit1.72 GB

70RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.01.06 GB

71RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.088 B

72ARG CACHEBUST=17936962310 B

73ARG ATOM_MESH_FEATURES=0 B

74RUN |23 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0814.03 MB

75RUN |23 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

76RUN |23 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.02.01 MB

77RUN |23 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.014.61 MB

78ARG LMCACHE_WHEEL_SHA256=68c0f6870c82497b4a1fef5c365960119bedfe66715a410457ae45eb6164cc030 B

79ARG LMCACHE_TAG=v0.4.50 B

80RUN |25 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.01.22 GB

81ARG INSTALL_AGENTX_HARNESS=10 B

82ARG AGENTX_HARNESS_VERSION=1.0.90 B

83RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.0302.38 MB

84RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.010.87 KB

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

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

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

88RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

89COPY /staging/ /opt/rocm/ # buildkit236 B

90RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.1.0 ROCM_TORCH_VERSION=2.13.032 B

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

Command

 ARG RELEASE