sha256:94fd001964e1cf55c3224a445b1fb5be31a7dac302315255db8422d813edd7f5
OS/ARCH
linux/amd64
Compressed size
8.59 GB
Last pushed
9 months by rocmdev
Type
Image
Manifest digest
sha256:94fd001964e1cf55c3224a445b1fb5be31a7dac302315255db8422d813edd7f5
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 B8ARG ROCM_VERSION=5.30 B9ARG AMDGPU_VERSION=5.30 B10ARG APT_PREF0 B11RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:237 B12RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:4.99 GB13RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:656 B14ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B15ENV ROCM_PATH=/opt/rocm0 B16ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B17ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx12010 B18ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx12010 B19ENV AITER_ROCM_ARCH=gfx942;gfx9500 B20ARG PYTHON_VERSION=3.120 B21RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh93 B22WORKDIR /app32 B23ENV DEBIAN_FRONTEND=noninteractive0 B24RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh110.59 MB25RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh65.68 MB26RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 /bin/sh2.06 GB27ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete0 B28ARG TRITON_BRANCH=57c693b60 B29ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B30ARG PYTORCH_BRANCH=1c57644d0 B31ARG PYTORCH_VISION_BRANCH=v0.23.00 B32ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B33ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git0 B34ARG FA_BRANCH=0e60e3940 B35ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B36ARG AITER_BRANCH=6a4e6e3580 B37ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B38RUN |13 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete351 B39COPY pytorch_afo_testkit/afo/gemm/benchmarks/afo_tune_device_*_full.csv /app # buildkit61.22 MB40ENV PYTORCH_TUNABLEOP_ENABLED=10 B41ENV PYTORCH_TUNABLEOP_TUNING=00 B42ENV PYTORCH_TUNABLEOP_FILENAME=/app/afo_tune_device_%d_full.csv0 B43ARG ARG_PYTORCH_ROCM_ARCH0 B44ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx12010 B45RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c103.67 MB46RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.35 KB47RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c214 B48ARG COMMON_WORKDIR=/app0 B49WORKDIR /app32 B50RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.97 KB51RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh32 B52RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh13.34 MB53RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.17 GB54ARG COMMON_WORKDIR=/app0 B55COPY /benchmarks /app/vllm/benchmarks # buildkit149.02 KB56COPY /examples /app/vllm/examples # buildkit173.17 KB57COPY /docker /app/vllm/docker # buildkit1.63 KB58ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B59ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B60ENV TOKENIZERS_PARALLELISM=false0 B61ENV SAFETENSORS_FAST_GPU=10 B62ENV HIP_FORCE_DEV_KERNARG=10 B63CMD ["/bin/bash"]0 BCommand
ARG RELEASE