sha256:1f8c1d83fb0c6ab9c6447db882356601baa1dda35a964c54eec33d06d0143f9b
OS/ARCH
linux/amd64
Compressed size
37.18 GB
Last pushed
10 months by genesu100
Type
Image
Manifest digest
sha256:b99b21b482627c691938a2625df7266f1e4e447996381207a6a7a579985c3ef4
Image Layers
1ARG RELEASE0 B2ARG LAUNCHPAD_BUILD_ARCH0 B3LABEL org.opencontainers.image.ref.name=ubuntu0 B4LABEL org.opencontainers.image.version=22.040 B5ADD file ... in / 28.17 MB6CMD ["/bin/bash"]0 B7ARG PYTHON_VERSION0 B8ARG PYTORCH_VERSION0 B9ARG ROCM_VERSION0 B10ARG AMDGPU_VERSION0 B11ENV DEBIAN_FRONTEND=noninteractive0 B12RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1279.21 KB13RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1111.82 MB14ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B15ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B16RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.196.99 MB17RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.17.96 MB18ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B19RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.116.68 KB20RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1390.62 KB21RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.16.45 GB22RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.13.69 GB23RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1120.57 KB24ENV ROCM_PATH=/opt/rocm-7.1.00 B25WORKDIR /workspace/99 B26RUN /bin/sh -c pip install690.91 KB27ENV MAX_JOBS=1280 B28ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B29ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B30RUN /bin/sh -c pip install10.3 GB31RUN /bin/sh -c pip uninstall51.86 MB32RUN /bin/sh -c pip uninstall891.46 KB33RUN /bin/sh -c pip uninstall221.57 KB34RUN /bin/sh -c apt install54.79 MB35RUN /bin/sh -c cd /opt/rocm/share/amd_smi1.1 MB36ARG LLVM_REPO=https://github.com/ROCm/llvm-project.git0 B37ARG LLVM_BRANCH=release/rocm-rel-7.10 B38RUN |2 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 /bin/sh1.13 GB39ARG HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git0 B40ARG HIPBLASLT_BRANCH=cbfc63b3f596a42eceeaba19716b0a1e8cb2e5870 B41RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git80.73 MB42RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git175 B43RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git4.81 GB44RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git185 B45RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git172 B46RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git196.21 KB47RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git196.21 KB48RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git162 B49ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B50ARG TRITON_BRANCH=1cc4a7c4aac921b7fa4658e162058632b7faed550 B51RUN |6 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.76 GB52ARG TORCHAUDIO_REPO=https://github.com/pytorch/audio0 B53ARG TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd0 B54RUN |8 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git704.65 KB55ARG APEX_REPO=https://github.com/rocm/apex.git0 B56ARG APEX_BRANCH=release/1.9.00 B57RUN |10 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git83.96 MB58ARG VISION_REPO=https://github.com/pytorch/vision0 B59ARG VISION_BRANCH=acccf86477759b2d3500f1ae1be065f7b1e409ec0 B60RUN |12 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.84 MB61ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B62ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B63ENV GPU_ARCHS=gfx942;gfx9500 B64RUN |14 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git90.3 MB65ARG AITER_REPO=https://github.com/rocm/aiter.git0 B66ARG AITER_BRANCH=7a41cca67187bd5f77c337765a1a289337901cef0 B67RUN |16 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.78 GB68ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B69ARG TE_BRANCH=0018b1bce0240ad53a6ac24a3b8052d9f067d34b0 B70ENV NVTE_USE_HIPBLASLT=10 B71ENV NVTE_FRAMEWORK=pytorch0 B72ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B73ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B74ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B75ENV NVTE_CK_USES_BWD_V3=10 B76ENV NVTE_CK_USES_FWD_V3=10 B77ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B78ENV NVTE_CK_HOW_V3_BF16_CVT=20 B79ENV NVTE_USE_ROCM=10 B80ENV NVTE_CK_FUSED_ATTN_PATH=/workspace/aiter/op_tests/cpp/mha0 B81RUN |18 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.49 GB82ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B83ARG FLUX_BRANCH=Flux_patch0 B84RUN |20 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git176.36 MB85ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B86ARG DLRM_BRANCH=main0 B87RUN |22 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git92.53 KB88ARG TORCHTITAN_REPO=https://github.com/AMD-AIG-AIMA/torchtitan.git0 B89ARG TORCHTITAN_BRANCH=dev/aiter_attn_random_input0 B90RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git129.91 MB91ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B92ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B93RUN |26 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git117.67 MB94ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B95ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B96RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.06 MB97RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git118.13 MB98RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git970.71 KB99RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.33 KB100RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.73 MB101RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git129.92 MB102RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.45 MB103RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git8.19 MB104ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B105ARG GROUPED_GEMM_BRANCH=rocm0 B106RUN |30 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git108.39 KB107ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B108ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B109ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B110ENV MAMBA_FORCE_BUILD=TRUE0 B111ENV HIP_ARCHITECTURES=gfx942,gfx9500 B112RUN |32 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git16.63 MB113ARG ROCSHMEM_REPO=https://github.com/ROCm/rocSHMEM.git0 B114ARG ROCSHMEM_BRANCH=e95360961dfa5575a49fe79abdf41ab9e4aeb5430 B115RUN |34 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git99.86 MB116ARG MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git0 B117ARG MEGATRON_BRANCH=feature/fsdp_optimization0 B118RUN |36 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git121.59 MB119ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B120ARG PRIMUS_BRANCH=release/v25.100 B121RUN |38 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git802.14 MB122ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B123ARG TURBO_COMMIT=docker-release-v25.100 B124ENV MAX_JOBS=1280 B125RUN |40 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git246.73 MB126RUN |40 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git30.85 MB127WORKDIR /workspace/32 B128RUN |40 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git2.18 MB129RUN |40 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.46 MB130ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B131ARG FBGEMM_COMMIT=main0 B132ENV BUILD_ROCM_VERSION=7.10 B133ENV FBGEMM_TBE_ROCM_HIP_BACKWARD_KERNEL=10 B134ENV ROCM_VERSION=710000 B135ENV HIPBLAS_V2=10 B136RUN |42 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.19 GB137ENV GPU_ARCHS=native0 B138RUN |42 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git326 B139RUN |42 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git335 B140RUN |42 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git127 B141ARG GIT_COMMIT_TAG=4f13adce424c80e257ec9f26aec731fc49540fdc0 B142RUN |43 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git185 B143CMD ["/usr/bin/bash"]0 BCommand
ARG RELEASE