rocm/amd-mlperf:flux1_training_6.0

Manifest digest

sha256:38cbbb041c232c8a7caa53bb8409b8c2061f1c2682de247861c98d343378b3a2

OS/ARCH

linux/amd64

Compressed size

16.73 GB

Last pushed

about 1 month by rocmdev

Type

Image

Manifest digest

sha256:38cbbb041c232c8a7caa53bb8409b8c2061f1c2682de247861c98d343378b3a2

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=22.040 B
4ADD file ... in / 28.36 MB
5CMD ["/bin/bash"]0 B
6ARG BASE_IMAGE=ubuntu:22.040 B
7ARG PYTHON_VERSION=3.100 B
8ARG PYTORCH_VERSION=2.10.00 B
9ARG TORCHVISION_VERSION=0.25.00 B
10ARG TORCHAUDIO_VERSION=2.10.00 B
11ARG TRITON_VERSION=3.6.00 B
12ARG APEX_VERSION=1.10.00 B
13ARG ROCM_VERSION=7.2.20 B
14ARG UBUNTU_VERSION=22.040 B
15ARG PYTORCH_WHEEL_INDEX=https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.2/0 B
16LABEL org.opencontainers.image.ref.name=rocm/pytorch:rocm7.2.2_ubuntu22.04_py3.10_pytorch_release_2.10.00 B
17ENV DEBIAN_FRONTEND=noninteractive0 B
18RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.0279.21 KB
19RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.0114.86 MB
20ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/opt/rocm/lib0 B
21RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.097.19 MB
22RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.0121.76 MB
23ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
24RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.07.24 GB
25RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.01.92 GB
26RUN |10 BASE_IMAGE=ubuntu:22.04 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.10.0133.19 KB
27WORKDIR /workspace99 B
28/bin/sh -c pip install pybind11646.99 KB
29/bin/sh -c pip install ninja90 B
30/bin/sh -c pip install packaging90 B
31/bin/sh -c /usr/bin/python3 -m pip1.58 MB
32WORKDIR /workspace/deps120 B
33ADD dir:e1e6dcc6c7a11f1805a0d637110050f3b865c4a690ee24f5269b8193b41fe1f4 in /workspace/deps/patches 542 B
34ARG MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e0 B
35|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c git108.96 MB
36|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c pip0 B
37|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c cd44.04 MB
38ENV PYTHONPATH=:/workspace/deps/megatron_lm0 B
39|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c pip210 B
40|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c git563.61 MB
41|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c cd1.24 GB
42|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c pip15.36 MB
43COPY file:ed5474c53a73989d7e21a1cf22bef6a17ec4cb03e3ace8a099ba3734fefe0d19 in requirements.txt 273 B
44|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c pip3181.78 MB
45|1 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e /bin/sh -c python49.24 MB
46ARG TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd0 B
47|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c1.82 GB
48|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c3.2 GB
49|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c328 B
50|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c1.27 MB
51|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c223.75 KB
52|2 MEGATRON_COMMIT=f312f7db1ec26b30831aae0fac023c6895a4ac1e TE_COMMIT=261eda6ab09e603d816af78d31ba793f643f28fd /bin/sh -c92.73 KB
53WORKDIR /workspace/code119 B
54COPY dir:fcab7f01d8528bed521417503f340030cb98b6ff68a9ec8a87d7dca125541ee5 in . 37.13 KB

Command

 ARG RELEASE