rocm/pytorch:rocm10.0_ubuntu24.04_py3.12_pytorch_release_2.12.0

Manifest digest

sha256:55bf8baa2a513b1c05bd256119fbc57a6ca64170e6cf5fe519b1cdd0c458cfd9

OS/ARCH

linux/amd64

Compressed size

18.8 GB

Last pushed

about 1 month by rocm

Type

Image

Manifest digest

sha256:55bf8baa2a513b1c05bd256119fbc57a6ca64170e6cf5fe519b1cdd0c458cfd9

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.version=24.040 B

4ADD file ... in / 28.37 MB

5CMD ["/bin/bash"]0 B

6ARG PYTHON_VERSION=3.120 B

7COPY .github/scripts/install_rocm_deps.sh /tmp/ # buildkit1.3 KB

8COPY .github/scripts/install_python.sh /tmp/ # buildkit1.05 KB

9RUN |1 PYTHON_VERSION=3.12 /bin/sh -c312.95 MB

10RUN |1 PYTHON_VERSION=3.12 /bin/sh -c7.94 MB

11ENV VIRTUAL_ENV=/opt/venv PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

12ARG ROCM_VERSION=10.0.00 B

13ARG AMDGPU_FAMILY=device-all0 B

14ARG RELEASE_TYPE=nightly0 B

15ARG PYTHON_VERSION=3.120 B

16ARG INDEX_URL=https://stable.repo.amd.com/rocm/whl-next0 B

17ARG EXTRA_INDEX_URL=0 B

18ARG PIN_EXPLICIT_VERSIONS=true0 B

19ARG TORCH_VERSION=2.12.0+rocm10.0.00 B

20ARG TORCHAUDIO_VERSION=2.11.0+rocm10.0.00 B

21ARG TORCHVISION_VERSION=0.27.0+rocm10.0.00 B

22ARG APEX_VERSION=1.12.0+rocm10.0.00 B

23ARG TRITON_VERSION0 B

24LABEL image.title=PyTorch + ROCm runtime0 B

25COPY .github/scripts/install_pytorch_wheels.py /tmp/ # buildkit5.82 KB

26COPY .github/scripts/detect_device_arches.py /tmp/ # buildkit3.23 KB

27RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=10.0.0 AMDGPU_FAMILY=device-all128.14 MB

28RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=10.0.0 AMDGPU_FAMILY=device-all18.34 GB

29RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=10.0.0 AMDGPU_FAMILY=device-all130 B

Command

 ARG RELEASE