Sign inSign up

rocm/amd-mlperf:llama31_8b_training_5.1

Manifest digest

sha256:ec9dd42eec4bd7bb738027333346227598f2297c7dab1fe715327dd2dd75725f

OS/ARCH

linux/amd64

Compressed size

12.08 GB

Last pushed

8 months by rocmdev

Type

Image

Manifest digest

sha256:ec9dd42eec4bd7bb738027333346227598f2297c7dab1fe715327dd2dd75725f

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
7ARG PYTHON_VERSION=3.100 B
8ARG ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel0 B
9ARG LKG_BUILD_NUM=910 B
10ARG ARTF_SERV=http://compute-artifactory.amd.com0 B
11ARG ARTF_PRE=http://compute-artifactory.amd.com/artifactory/list0 B
12ARG PYTORCH_VERSION=2.8.00 B
13ARG ROCM_VERSION0 B
14ARG PYTHON_VERSION_PYTORCH=3100 B
15ARG PY_POST_COMP=0 B
16ARG WHL_INSTALL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-dkms-component-baas-rel/91/preview/torch-2.8.0.dev20251001+rocm7.0.0.lw.git64359f59-cp310-cp310-linux_x86_64.sh0 B
17ARG OS_VERSION=22.040 B
18ARG PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
19ENV DEBIAN_FRONTEND=noninteractive0 B
20RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91279.21 KB
21RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91110.57 MB
22ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
23ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B
24RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91102.1 MB
25RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=917.91 MB
26ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
27RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=913.47 GB
28RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=911.52 MB
29COPY /pytorch/whl_docker/custom_install.sh /var/lib/jenkins/custom_install.sh # buildkit1.5 KB
30COPY src_repo /var/lib/jenkins/pytorch # buildkit1.15 GB
31RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91211 B
32ENV BUILD_ENVIRONMENT=rocm0 B
33ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
34RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91780.76 MB
35RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=912.8 GB
36RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=91104.78 KB
37WORKDIR /workspace98 B
38/bin/sh -c pip install pybind11567.76 KB
39/bin/sh -c pip install ninja92 B
40/bin/sh -c pip install packaging92 B
41/bin/sh -c /usr/bin/python3 -m pip1.58 MB
42WORKDIR /workspace/deps119 B
43ADD dir:90e18f0e6005244189b5b0f7647bd7ed1911d895d51f3aac3cc6de7dbb49f054 in /workspace/deps/patches 412 B
44ARG MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb10 B
45|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c git86.25 MB
46|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c pip0 B
47|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c cd47.03 MB
48ENV PYTHONPATH=:/workspace/deps/megatron_lm0 B
49|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c git538.11 MB
50|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c cd590.36 MB
51|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c pip16.64 MB
52COPY file:30cc731d00832afe03b6f961f2dd0d3f835e7d52bda027d54765ff518ef91839 in requirements.txt 349 B
53|1 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 /bin/sh -c pip3216.44 MB
54ARG TE_COMMIT=f3692b67613849d27f0e04d6087dedf9402e208c0 B
55|2 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 TE_COMMIT=f3692b67613849d27f0e04d6087dedf9402e208c /bin/sh -c2.19 GB
56WORKDIR /workspace/code118 B
57COPY dir:ec5445c58d354803888bc15880153a16cca33cd24c4832258dde50ad994644f3 in . 16.9 KB
58|2 MEGATRON_COMMIT=5c34c4cfe41dc304ff51c686358cf039621c7cb1 TE_COMMIT=f3692b67613849d27f0e04d6087dedf9402e208c /bin/sh -c259 B

Command

 ARG RELEASE