Sign inSign up

rocm/amd-mlperf:llama2_70b_training_5.1

Manifest digest

sha256:568de7713c8af69f853e3d087b2377805d77527ed71fdea3f95f7666046c8519

OS/ARCH

linux/amd64

Compressed size

12.1 GB

Last pushed

8 months by rocmdev

Type

Image

Manifest digest

sha256:568de7713c8af69f853e3d087b2377805d77527ed71fdea3f95f7666046c8519

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
43ARG MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f854120 B
44|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c git86.14 MB
45|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c pip0 B
46|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c cd111.21 MB
47ENV PYTHONPATH=:/workspace/deps/megatron_lm0 B
48|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c git537.51 MB
49|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c cd511.96 MB
50COPY file:4d84055589fe1064bc439d883bcb16c2080840b6d73031c88923c84ed2ae3ed1 in requirements.txt 352 B
51|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c pip3216.84 MB
52|1 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 /bin/sh -c python49.04 MB
53ARG TE_COMMIT=d0ef413b5bfe9ea33b1c8626e99fb46967a607120 B
54|2 MEGATRON_COMMIT=bc0be960c2f1fb2c4a57ecd7dec6235c79f85412 TE_COMMIT=d0ef413b5bfe9ea33b1c8626e99fb46967a60712 /bin/sh -c2.18 GB
55WORKDIR /workspace/code118 B
56COPY dir:05874ff767814352393c8cb320b49a1b07671e5f5193299ec8a626e010e12de6 in . 726.86 KB

Command

 ARG RELEASE