mi355x_llama3_1_405b_inference_5.1
sha256:1551c365ade880b5cb38033d258b512f753c032eb31727fb0f0cb2bace26cb6f
OS/ARCH
linux/amd64
Compressed size
51.9 GB
Last pushed
10 months by rocmdev
Type
Image
Manifest digest
sha256:1551c365ade880b5cb38033d258b512f753c032eb31727fb0f0cb2bace26cb6f
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.16 MB6CMD ["/bin/bash"]0 B7ARG UBUNTU_VERSION=22.040 B8ENV DEBIAN_FRONTEND=noninteractive0 B9ARG PYTORCH_ROCM_ARCH=gfx942;gfx9500 B10ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B11COPY ./common/install_base.sh install_base.sh # buildkit1.85 KB12RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 /bin/sh301.02 MB13COPY ./common/install_user.sh install_user.sh # buildkit703 B14RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 /bin/sh1.21 KB15ARG KATEX=0 B16COPY ./common/install_docs_reqs.sh install_docs_reqs.sh # buildkit483 B17RUN |3 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=110 B18ARG ANACONDA_PYTHON_VERSION=3.100 B19ARG BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.4-py3.100 B20ENV ANACONDA_PYTHON_VERSION=3.100 B21ENV PATH=/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B22ARG CONDA_CMAKE=yes0 B23COPY requirements-ci.txt /opt/conda/requirements-ci.txt # buildkit3.7 KB24COPY ./common/install_conda.sh install_conda.sh # buildkit1.87 KB25COPY ./common/common_utils.sh common_utils.sh # buildkit700 B26RUN |6 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=2.55 GB27ARG GCC_VERSION=0 B28COPY ./common/install_gcc.sh install_gcc.sh # buildkit378 B29RUN |7 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=104 B30ARG CLANG_VERSION=0 B31COPY ./common/install_clang.sh install_clang.sh # buildkit791 B32RUN |8 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=106 B33ARG PROTOBUF=yes0 B34COPY ./common/install_protobuf.sh install_protobuf.sh # buildkit503 B35RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=96.43 MB36RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=109 B37ENV INSTALLED_PROTOBUF=yes0 B38ARG VISION=yes0 B39COPY ./common/install_vision.sh ./common/cache_vision_models.sh ./common/common_utils.sh ./1.44 KB40RUN |10 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=427.52 MB41RUN |10 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=159 B42ENV INSTALLED_VISION=yes0 B43ARG ROCM_VERSION=6.40 B44COPY ./common/install_rocm.sh install_rocm.sh # buildkit2.04 KB45RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=5.72 GB46RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=105 B47COPY ./common/install_rocm_magma.sh install_rocm_magma.sh # buildkit1.09 KB48RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=345.49 MB49RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=111 B50ADD ./common/install_miopen.sh install_miopen.sh # buildkit1.52 KB51RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=107 B52ENV ROCM_PATH=/opt/rocm0 B53ENV PATH=/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B54ENV PATH=/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B55ENV PATH=/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B56ENV PATH=/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B57ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B58ENV MAGMA_HOME=/opt/rocm/magma0 B59ENV LANG=C.UTF-80 B60ENV LC_ALL=C.UTF-80 B61COPY ./common/install_amdsmi.sh install_amdsmi.sh # buildkit166 B62RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=2.73 MB63RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=107 B64ARG UCX_COMMIT=0 B65ARG UCC_COMMIT=0 B66ENV UCX_COMMIT=0 B67ENV UCC_COMMIT=0 B68ENV UCX_HOME=/usr0 B69ENV UCC_HOME=/usr0 B70ADD ./common/install_ucc.sh install_ucc.sh # buildkit801 B71RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=32 B72RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=104 B73COPY ./common/install_openssl.sh install_openssl.sh # buildkit425 B74ENV OPENSSL_ROOT_DIR=/opt/openssl0 B75RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=18.41 MB76ENV OPENSSL_DIR=/opt/openssl0 B77ARG INDUCTOR_BENCHMARKS=0 B78ARG ANACONDA_PYTHON_VERSION=3.100 B79ENV ANACONDA_PYTHON_VERSION=3.100 B80COPY ./common/install_inductor_benchmark_deps.sh install_inductor_benchmark_deps.sh # buildkit642 B81COPY ./common/common_utils.sh common_utils.sh # buildkit700 B82COPY ci_commit_pins/huggingface.txt huggingface.txt # buildkit141 B83COPY ci_commit_pins/timm.txt timm.txt # buildkit136 B84RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=32 B85RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=190 B86ARG CMAKE_VERSION=3.18.50 B87COPY ./common/install_cmake.sh install_cmake.sh # buildkit565 B88RUN |15 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=41.21 MB89RUN |15 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=106 B90ARG NINJA_VERSION=1.9.00 B91COPY ./common/install_ninja.sh install_ninja.sh # buildkit345 B92RUN |16 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=86.22 KB93RUN |16 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=106 B94ARG TRITON=yes0 B95COPY ./common/install_triton.sh install_triton.sh # buildkit1.37 KB96COPY ./common/common_utils.sh common_utils.sh # buildkit700 B97COPY ci_commit_pins/triton.txt triton.txt # buildkit137 B98COPY triton_version.txt triton_version.txt # buildkit120 B99RUN |17 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=4.93 GB100RUN |17 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=172 B101COPY ./common/install_cache.sh install_cache.sh # buildkit1.88 KB102ENV PATH=/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B103RUN |17 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=15.24 MB104COPY ./common/install_openmpi.sh install_openmpi.sh # buildkit304 B105RUN |17 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=32 B106RUN |17 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx942;gfx950 KATEX=108 B107ARG BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.4-py3.100 B108ENV BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.4-py3.100 B109COPY /opt/llvm /opt/llvm # buildkit51.64 MB110USER jenkins0 B111CMD ["bash"]0 B112ARG FRAME_NAME=pytorch0 B113ARG OS_CHOICE=ubuntu22.040 B114ARG PY_VER=py3.100 B115ARG PY_EXTRA_PK=0 B116ARG MAX_JOBS=320 B117ENV MAX_JOBS=320 B118USER root0 B119COPY ./pytorch-extra /var/lib/jenkins # buildkit6.96 KB120COPY ./ubuntu22.04-pytorch /var/lib/jenkins/pytorch # buildkit3 GB121WORKDIR /var/lib/jenkins32 B122RUN |5 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_VER=py3.1032 B123RUN |5 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_VER=py3.102.7 GB124ENV ROCM_PATH=/opt/rocm0 B125ENV MPI_HOME=/opt/ompi0 B126ENV UCX_HOME=/opt/ucx0 B127ENV PATH=/opt/ompi/bin:/opt/ucx/bin:/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B128ENV LD_LIBRARY_PATH=/opt/ompi/lib:/opt/rocm/lib:/usr/local/lib:0 B129ENV CI=10 B130ENV PYTORCH_TEST_WITH_ROCM=10 B131ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B132ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B133ARG ARG_PYTORCH_ROCM_ARCH=gfx942;gfx9500 B134ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B135RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 /bin/sh -c93.11 MB136RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 /bin/sh -c4.97 MB137RUN |1 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 /bin/sh -c294 B138ARG COMMON_WORKDIR=/app0 B139WORKDIR /app92 B140RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 COMMON_WORKDIR=/app /bin/sh2.09 KB141RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 COMMON_WORKDIR=/app /bin/sh32 B142RUN |2 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 COMMON_WORKDIR=/app /bin/sh2.77 MB143ARG BUILD_RPD=10 B144RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 COMMON_WORKDIR=/app BUILD_RPD=17.98 MB145RUN |3 ARG_PYTORCH_ROCM_ARCH=gfx942;gfx950 COMMON_WORKDIR=/app BUILD_RPD=1862.44 MB146ARG COMMON_WORKDIR=/app0 B147COPY /benchmarks /app/vllm/benchmarks # buildkit105.81 KB148COPY /examples /app/vllm/examples # buildkit100.12 KB149ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B150ENV TOKENIZERS_PARALLELISM=false0 B151ENV HIP_FORCE_DEV_KERNARG=10 B152CMD ["/bin/bash"]0 B153/bin/bash305.05 MB154/bin/bash1.07 GB155/bin/bash4.78 GB156/bin/bash2.98 GB157/bin/bash4.93 GB158/bin/bash4.34 MB159/bin/bash347.46 KB160/bin/bash206.48 MB161/bin/bash500.75 KB162/bin/bash853.07 MB163/bin/bash3.24 GB164/bin/bash19.81 MB165/bin/bash16.54 MB166/bin/bash2.05 GB167/bin/bash39.43 MB168/bin/bash4.66 GB169/bin/bash532.87 MB170/bin/bash92.22 MB171bash306.3 MB172bash42.23 MB173bash3.53 GB174bash8.27 MB175bash3.43 MB176bash652.91 MB177bash117.32 KB178bash1.71 MB179bash0 B180bash374.07 MB181bash79.64 MB182bash46.56 MBCommand
ARG RELEASE