Sign inSign up

rocm/verl:verl-0.6.0.amd0_rocm7.0_vllm0.11.0.dev

Manifest digest

sha256:f70a3ebc94c1f66de42a2fcc3f8a6a8d6d0881eb0e65b6958d7d6d24b3eecb0d

OS/ARCH

linux/amd64

Compressed size

8.9 GB

Last pushed

8 months by rocmdev

Type

Image

Manifest digest

sha256:f70a3ebc94c1f66de42a2fcc3f8a6a8d6d0881eb0e65b6958d7d6d24b3eecb0d

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 29.04 MB
6CMD ["/bin/bash"]0 B
7LABEL [email protected]0 B
8ARG ROCM_VERSION=5.30 B
9ARG AMDGPU_VERSION=5.30 B
10ARG APT_PREF0 B
11RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:237 B
12RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:4.99 GB
13RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:656 B
14ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15ENV ROCM_PATH=/opt/rocm0 B
16ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B
17ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
18ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
19ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
20ARG PYTHON_VERSION=3.120 B
21RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh93 B
22WORKDIR /app32 B
23ENV DEBIAN_FRONTEND=noninteractive0 B
24RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh112.15 MB
25RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh65.7 MB
26RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh2.1 GB
27ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete0 B
28ARG TRITON_BRANCH=57c693b60 B
29ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B
30ARG PYTORCH_BRANCH=1c57644d0 B
31ARG PYTORCH_VISION_BRANCH=v0.23.00 B
32ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B
33ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git0 B
34ARG FA_BRANCH=0e60e3940 B
35ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B
36ARG AITER_BRANCH=9716b1b80 B
37ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
38RUN |13 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete350 B
39COPY pytorch_afo_testkit/afo/gemm/benchmarks/afo_tune_device_*_full.csv /app # buildkit61.21 MB
40ENV PYTORCH_TUNABLEOP_ENABLED=10 B
41ENV PYTORCH_TUNABLEOP_TUNING=00 B
42ENV PYTORCH_TUNABLEOP_FILENAME=/app/afo_tune_device_%d_full.csv0 B
43RUN apt-get update -q -y113.28 MB
44RUN python3 -m pip install5.19 MB
45RUN apt-get purge -y sccache;244 B
46RUN python3 -m pip install2.59 KB
47RUN python3 -m pip install12.42 MB
48RUN --mount=type=bind,from=export_vllm,src=/,target=/install 1.26 GB
49COPY --from=export_vllm /benchmarks ${COMMON_WORKDIR}/vllm/benchmarks164.46 KB
50COPY --from=export_vllm /examples ${COMMON_WORKDIR}/vllm/examples198.66 KB
51RUN pip install "tensordict==0.6.2" --no-deps47.71 MB
52WORKDIR /workspace/224 B
53RUN git clone ${VERL_REPO} &&115.74 MB

Command

 ARG RELEASE