sha256:f5c749ebdcaee99dd9a2b325f868a78c93278d0169af5b8df597fe0413aa72e2
OS/ARCH
linux/amd64
Compressed size
35.79 GB
Last pushed
about 12 hours by valarlip
Type
Image
Manifest digest
sha256:f5c749ebdcaee99dd9a2b325f868a78c93278d0169af5b8df597fe0413aa72e2
Image Layers
1ARG RELEASE0 B2ARG LAUNCHPAD_BUILD_ARCH0 B3LABEL org.opencontainers.image.version=24.040 B4ADD file ... in / 28.37 MB5CMD ["/bin/bash"]0 B6ARG PYTHON_VERSION=3.120 B7COPY .github/scripts/install_rocm_deps.sh /tmp/ # buildkit1.3 KB8COPY .github/scripts/install_python.sh /tmp/ # buildkit1.05 KB9RUN |1 PYTHON_VERSION=3.12 /bin/sh -c312.94 MB10RUN |1 PYTHON_VERSION=3.12 /bin/sh -c8.61 MB11ENV VIRTUAL_ENV=/opt/venv PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B12ARG ROCM_VERSION=7.14.00 B13ARG AMDGPU_FAMILY=device-all0 B14ARG RELEASE_TYPE=nightly0 B15ARG PYTHON_VERSION=3.120 B16ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch0 B17ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/0 B18ARG PIN_EXPLICIT_VERSIONS=true0 B19ARG TORCH_VERSION=2.13.0+rocm7.14.00 B20ARG TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.00 B21ARG TORCHVISION_VERSION=0.28.0+rocm7.14.00 B22ARG APEX_VERSION0 B23ARG TRITON_VERSION0 B24LABEL image.title=PyTorch + ROCm runtime0 B25COPY .github/scripts/install_pytorch_wheels.py /tmp/ # buildkit5.85 KB26COPY .github/scripts/detect_device_arches.py /tmp/ # buildkit3.32 KB27RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all127.98 MB28RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all17.66 GB29RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all130 B30ARG PYTHON_VERSION=3.120 B31ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch0 B32ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/0 B33ARG AMDGPU_FAMILY=device-all0 B34ARG ROCM_VERSION=7.14.00 B35ARG PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx9500 B36ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_SOURCE_DIR=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_BIN=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin0 B37ENV CMAKE_PREFIX_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/cmake HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_SYSDEPS_INCLUDE=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/include0 B38RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all1009.99 MB39RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all6.81 GB40COPY pytorch /workspace/pytorch # buildkit130.79 MB41RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all176 B42RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all1.43 GB43RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all92 B44RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all217 B45RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all135.05 KB46WORKDIR /0 B47ARG GPU_ARCH=gfx942;gfx9500 B48ENV GPU_ARCH_LIST=gfx942;gfx9500 B49ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B50RUN |1 GPU_ARCH=gfx942;gfx950 /bin/sh -c124.3 MB51RUN |1 GPU_ARCH=gfx942;gfx950 /bin/sh -c12.41 KB52ARG ATOM_REPO=https://github.com/ROCm/ATOM.git0 B53ARG ATOM_COMMIT=main0 B54RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main139.31 MB55RUN |3 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main14.08 MB56ARG INSTALL_MOONCAKE=10 B57ARG MOONCAKE_REPO=https://github.com/Jasen2201/Mooncake.git0 B58ARG MOONCAKE_COMMIT=fix/ionic-mr-and-qp-resource-fixes0 B59ARG VENV_PYTHON=/opt/venv/bin/python0 B60RUN |7 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main92.25 MB61ENV PATH=/usr/local/go/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B62RUN |7 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main212.84 MB63ARG 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.deb0 B64RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main343.6 KB65RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main319.56 MB66ARG RUST_VERSION=1.94.00 B67RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main203.15 MB68ENV PATH=/root/.cargo/bin:/usr/local/go/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B69COPY /app/rccl/build/release/*.deb /tmp/rccl/ # buildkit21.69 MB70RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main23.18 MB71COPY /app/aiter-test /app/aiter-test # buildkit1.65 GB72RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main972.78 MB73RUN |9 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main92 B74ARG CACHEBUST=17864577740 B75RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main681.41 MB76RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main32 B77RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main1001.18 KB78RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main2.86 MB79RUN |10 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main16.18 MB80ARG LMCACHE_TAG=v0.4.50 B81RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main69.41 MB82ARG INSTALL_SA_AIPERF=10 B83ARG SA_AIPERF_COMMIT=0d2aa0572ac685943d38c580675c4a61023581d30 B84RUN |13 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=main202.02 MB85CMD ["/bin/bash"]0 B86ARG MAX_JOBS=640 B87ARG VENV_PYTHON=/opt/venv/bin/python0 B88ARG VLLM_REPO=https://github.com/vllm-project/vllm.git0 B89ARG VLLM_COMMIT=752a3a504485790a2e8491cacbb35c137339ad340 B90ARG INSTALL_LM_EVAL=10 B91ARG INSTALL_FASTSAFETENSORS=10 B92LABEL com.rocm.atom.vllm_commit=752a3a504485790a2e8491cacbb35c137339ad340 B93ENV PATH=/opt/venv/bin:/root/.cargo/bin:/usr/local/go/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B94ENV MAX_JOBS=640 B95ENV VLLM_TARGET_DEVICE=rocm0 B96ENV CMAKE_MAKE_PROGRAM=/usr/local/bin/ninja0 B97RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git22.34 MB98RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git59.05 MB99RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git423.18 MB100COPY docker/patches/vllm-torch210-compat.patch /tmp/vllm-torch210-compat.patch # buildkit2.07 KB101RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git292.09 MB102RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git613.24 MB103RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git415.13 KB104RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git1.75 GB105RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git80.08 MB106RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git92 B107RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git423.19 MB108RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git22.36 MB109RUN |6 MAX_JOBS=64 VENV_PYTHON=/opt/venv/bin/python VLLM_REPO=https://github.com/vllm-project/vllm.git32 B110CMD ["/bin/bash"]0 BCommand
ARG RELEASE