Sign inSign up

rocm/primus:v26.3

Index digest

sha256:1a02b74a94d82131f3f119a94ccbed45bb5b4cd77a0616cb4e6a29e64a048482

OS/ARCH

linux/amd64

Compressed size

21.05 GB

Last pushed

3 months by rocmdev

Type

Image

Manifest digest

sha256:da50bfe9dc4bb70ad683d3ee4a176dccb66f596f48d30e8e52323b41892759b1

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
6ENV DEBIAN_FRONTEND=noninteractive0 B
7RUN /bin/sh -c apt update500.3 MB
8RUN /bin/sh -c wget https://repo.radeon.com/amdgpu-install/7.2.1/ubuntu/noble/amdgpu-install_7.2.1.70201-1_all.deb7.41 GB
9ENV ROCM_PATH=/opt/rocm-7.2.10 B
10WORKDIR /workspace/99 B
11ENV MAX_JOBS=1280 B
12ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
13ENV PATH=/opt/rocm-7.2.1/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
14ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B
15RUN /bin/sh -c python -m4 MB
16ENV PATH=/opt/venv/bin:/opt/rocm-7.2.1/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
17RUN /bin/sh -c pip install33.24 MB
18RUN /bin/sh -c apt update32 B
19ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
20ENV HSA_NO_SCRATCH_RECLAIM=10 B
21RUN /bin/sh -c pip uninstall165.99 MB
22ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B
23ARG PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d0 B
24RUN |2 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d /bin/sh460.89 MB
25ARG APEX_REPO=https://github.com/ROCm/apex.git0 B
26ARG APEX_BRANCH=e37ed124b424178962348bdb4c2a06a81766cadb0 B
27RUN |4 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git132.02 MB
28RUN |4 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git1.26 MB
29ARG HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git0 B
30ARG HIPBLASLT_BRANCH=c4b2dc9869497e664e02b110288d158375438e740 B
31RUN |6 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git21.62 MB
32RUN |6 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git176 B
33RUN |6 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git1.69 GB
34RUN |6 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git184 B
35RUN |6 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git203.43 KB
36ARG TORCHAUDIO_REPO=https://github.com/pytorch/audio0 B
37ARG TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd0 B
38RUN |8 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git831.14 KB
39ARG VISION_REPO=https://github.com/pytorch/vision0 B
40ARG VISION_BRANCH=acccf86477759b2d3500f1ae1be065f7b1e409ec0 B
41RUN |10 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git2.22 MB
42ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
43ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
44ENV GPU_ARCHS=gfx942;gfx9500 B
45RUN |12 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git89.5 MB
46ARG AITER_REPO=https://github.com/rocm/aiter.git0 B
47ARG AITER_BRANCH=f299f579a63b8de4f008e19b6f8867dbfa9eaf390 B
48RUN |14 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git2.51 GB
49ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
50ARG TE_BRANCH=40434cf6def378bfd8693754e87bcc430f95695a0 B
51ENV NVTE_USE_HIPBLASLT=10 B
52ENV NVTE_FRAMEWORK=pytorch0 B
53ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B
54ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
55ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
56ENV NVTE_CK_USES_BWD_V3=10 B
57ENV NVTE_CK_USES_FWD_V3=10 B
58ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
59ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
60ENV NVTE_USE_ROCM=10 B
61ENV NVTE_CK_FUSED_ATTN_PATH=/workspace/aiter/op_tests/cpp/mha0 B
62RUN |16 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git4.29 GB
63ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
64ARG FLUX_BRANCH=Flux_patch0 B
65RUN |18 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git103.82 MB
66ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
67ARG DLRM_BRANCH=main0 B
68RUN |20 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git103.6 KB
69ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
70ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
71RUN |22 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git42.6 MB
72ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
73ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
74RUN |24 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git3.62 MB
75RUN |24 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git207.39 MB
76ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
77ARG GROUPED_GEMM_BRANCH=rocm0 B
78RUN |26 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git112.78 KB
79ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
80ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
81ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
82ENV MAMBA_FORCE_BUILD=TRUE0 B
83ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
84RUN |28 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git8.45 MB
85ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
86ARG MAMBA_BRANCH=4d67c534d10b7299194ede55b36718cc7a1dd4720 B
87RUN |30 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git96.42 MB
88ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
89ARG PRIMUS_BRANCH=45955d457d7d083cc6bcd9a669b97249ed71c4560 B
90ENV NVTE_FLASH_ATTN=00 B
91ENV NVTE_FUSED_ATTN=10 B
92RUN |32 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git1.01 GB
93ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
94ARG TURBO_COMMIT=3cd482def8bb4814b1799fe481555559fff040b80 B
95ENV MAX_JOBS=1280 B
96ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
97RUN |34 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git606.86 MB
98RUN |34 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git32.44 MB
99WORKDIR /workspace/32 B
100RUN |34 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git5.01 MB
101ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
102ARG FBGEMM_COMMIT=fbda30f767186b7cc6f5663fd17c268a5d853c3e0 B
103ENV BUILD_ROCM_VERSION=7.20 B
104ENV FBGEMM_TBE_ROCM_HIP_BACKWARD_KERNEL=10 B
105ENV ROCM_VERSION=720000 B
106ENV HIPBLAS_V2=10 B
107RUN |36 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git1.32 GB
108RUN |36 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git32 B
109ENV GPU_ARCHS=native0 B
110RUN |36 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git7.9 MB
111RUN |36 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git1.54 MB
112ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
113RUN |37 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git703.12 KB
114RUN |37 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git232 B
115ARG UCX_VERSION=1.18.00 B
116RUN |38 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git87.65 MB
117ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
118ARG MPI_VERSION=4.1.60 B
119RUN |39 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git3.85 MB
120ARG RCCL_REPO=https://github.com/ROCm/rccl.git0 B
121ARG RCCL_BRANCH=release/rocm-rel-7.20 B
122RUN |41 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git241.75 MB
123ARG ANP_REPO=https://github.com/rocm/amd-anp.git0 B
124ARG ANP_BRANCH=tags/v1.3.00 B
125ENV RCCL_HOME=/workspace/rccl0 B
126ENV MPI_PATH=/ompi-4.1.6/install0 B
127RUN |43 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git32 B
128RUN |43 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git578.44 KB
129ARG ROCSHMEM_REPO=https://github.com/ROCm/rocSHMEM.git0 B
130ARG ROCSHMEM_BRANCH=17ff985c026f9f97f85068647e863ab541dd56450 B
131ENV ROCSHMEM_HOME=/opt/rocshmem0 B
132ENV UCX_HOME=/opt/ucx0 B
133ENV MPI_HOME=/usr/lib/x86_64-linux-gnu/openmpi0 B
134RUN |45 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git4.9 MB
135RUN |45 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git258 B
136RUN |45 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git454 B
137RUN |45 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git32 B
138RUN |45 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git225 B
139ARG GIT_COMMIT_TAG=428dc232095894b4c89cad237d301057462bb5290 B
140ARG DOCKERFILE_PATH=primus_training/docker/Dockerfile0 B
141RUN |47 PYTORCH_REPO=https://github.com/ROCm/pytorch.git PYTORCH_BRANCH=94c6e047c2aa9f11c4cca60b4b90ea57b0eec54d APEX_REPO=https://github.com/ROCm/apex.git32.63 KB
142COPY primus_training/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit7.36 KB
143CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE