Sign inSign up

rocm/primus:pytorch-2.12.0-rocm7.14.0a20260608_te-2.14.0.dev0-e6ede467_v26.4

Index digest

sha256:8c8ecc6fe14b5061423cc82517831e3515f6eb15647ee6d746fd6a0c1963da24

OS/ARCH

linux/amd64

Compressed size

16.25 GB

Last pushed

about 2 months by rocmdev

Type

Image

Manifest digest

sha256:c3d9033c55440f3650a24b26377f3630a0b1bf1b48a6509a4b8fdd8900b2bbcb

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
6WORKDIR /workspace/97 B
7ENV MAX_JOBS=1280 B
8ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
9ENV ROCM_AMDGPU_TARGETS=gfx942,gfx9500 B
10ENV DEBIAN_FRONTEND=noninteractive0 B
11RUN /bin/sh -c apt update274.45 MB
12RUN /bin/sh -c python -m4 MB
13ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
14RUN /bin/sh -c pip install33.32 MB
15RUN /bin/sh -c apt update137.51 MB
16ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
17ENV HSA_NO_SCRATCH_RECLAIM=10 B
18ARG PYTORCH_VERSION=2.12.0+rocm7.14.0a202606080 B
19RUN |1 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 /bin/sh -c73.44 MB
20RUN |1 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 /bin/sh -c3.74 GB
21RUN |1 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 /bin/sh -c877.07 MB
22ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
23ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
24ENV HIP_PLATFORM=amd0 B
25ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
27ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
28ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
29ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
30ENV PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
31ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib0 B
32ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib640 B
33ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
34ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
35ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
36ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
37ENV GPU_ARCHS=gfx942;gfx9500 B
38RUN |3 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d880.77 MB
39ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
40ARG TE_BRANCH=e6ede467a49cfda1859b145e045109e2f330bccc0 B
41ENV NVTE_USE_HIPBLASLT=10 B
42ENV NVTE_FRAMEWORK=pytorch0 B
43ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B
44ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
45ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
46ENV NVTE_CK_USES_BWD_V3=10 B
47ENV NVTE_CK_USES_FWD_V3=10 B
48ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
49ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
50ENV NVTE_USE_ROCM=10 B
51RUN |5 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d86.85 GB
52ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
53ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
54RUN |7 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8104.49 MB
55ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
56ARG DLRM_BRANCH=main0 B
57RUN |9 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8103.6 KB
58ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
59ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
60RUN |11 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d821.6 MB
61ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
62ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
63RUN |13 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.61 MB
64RUN |13 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8206.75 MB
65ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
66ARG GROUPED_GEMM_BRANCH=rocm0 B
67RUN |15 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8112.5 KB
68ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
69ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
70ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
71ENV MAMBA_FORCE_BUILD=TRUE0 B
72ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
73RUN |17 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d88.89 MB
74ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
75ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
76RUN |19 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8270.15 MB
77ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
78ARG PRIMUS_BRANCH=236cfa9d8764bf8519063178dcc6279c635c6f6c0 B
79ENV NVTE_FLASH_ATTN=00 B
80ENV NVTE_FUSED_ATTN=10 B
81RUN |21 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.11 GB
82ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
83ARG AITER_COMMIT=b5e03ed191fca11ee423226537ef8d9435e432a60 B
84RUN |23 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8226.45 MB
85ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
86ARG TURBO_COMMIT=3c39ef259aa6d724c77c481e926466e7a167e9380 B
87ENV MAX_JOBS=1280 B
88ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
89RUN |25 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8429.94 MB
90RUN |25 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d813.36 MB
91WORKDIR /workspace/32 B
92RUN |25 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d85.19 MB
93ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
94ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
95ENV BUILD_ROCM_VERSION=7.140 B
96RUN |27 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8975.23 KB
97RUN |27 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.7 GB
98RUN |27 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8377.33 KB
99ENV GPU_ARCHS=native0 B
100RUN |27 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d88.08 MB
101RUN |27 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d811.06 MB
102ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
103RUN |28 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d82.74 MB
104RUN |28 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8233 B
105ARG UCX_VERSION=1.18.00 B
106RUN |29 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d886.43 MB
107ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
108ARG MPI_VERSION=4.1.60 B
109RUN |30 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.84 MB
110ENV PATH=/opt/openmpi/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
111ENV LD_LIBRARY_PATH=/opt/openmpi/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib0 B
112ARG ROCSHMEM_REPO=https://github.com/ROCm/rocSHMEM.git0 B
113ARG ROCSHMEM_BRANCH=17ff985c026f9f97f85068647e863ab541dd56450 B
114ENV ROCSHMEM_HOME=/opt/rocshmem0 B
115ENV UCX_HOME=/workspace/ucx-1.18.0/install0 B
116ENV MPI_HOME=/opt/openmpi0 B
117RUN |32 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d84.57 MB
118RUN |32 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8259 B
119RUN |32 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8451 B
120RUN |32 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d832 B
121RUN |32 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8146 B
122ARG GIT_COMMIT_TAG=a2d4c7239d874501f864080baf2a8131236b02210 B
123ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
124RUN |34 PYTORCH_VERSION=2.12.0+rocm7.14.0a20260608 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d823 KB
125COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit4.98 KB
126CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE