Sign inSign up

rocm/pytorch:latest

Manifest digest

sha256:a3867e22ca44c458b1feda92026c6dc289e7e447093cf55914d1fd56ef6d7372

OS/ARCH

linux/amd64

Compressed size

27.47 GB

Last pushed

6 days by rocm

Type

Image

Manifest digest

sha256:a3867e22ca44c458b1feda92026c6dc289e7e447093cf55914d1fd56ef6d7372

Image Layers

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.94 MB
10RUN |1 PYTHON_VERSION=3.12 /bin/sh -c8.61 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=7.14.00 B
13ARG AMDGPU_FAMILY=device-all0 B
14ARG RELEASE_TYPE=nightly0 B
15ARG PYTHON_VERSION=3.120 B
16ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch0 B
17ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/0 B
18ARG PIN_EXPLICIT_VERSIONS=true0 B
19ARG TORCH_VERSION=2.13.0+rocm7.14.00 B
20ARG TORCHAUDIO_VERSION=2.11.0.2+rocm7.14.00 B
21ARG TORCHVISION_VERSION=0.28.0+rocm7.14.00 B
22ARG APEX_VERSION0 B
23ARG TRITON_VERSION0 B
24LABEL image.title=PyTorch + ROCm runtime0 B
25COPY .github/scripts/install_pytorch_wheels.py /tmp/ # buildkit5.85 KB
26COPY .github/scripts/detect_device_arches.py /tmp/ # buildkit3.32 KB
27RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all127.98 MB
28RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all17.66 GB
29RUN |12 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all130 B
30ARG PYTHON_VERSION=3.120 B
31ARG INDEX_URL=https://rocm.devreleases.amd.com/whl-multi-arch0 B
32ARG EXTRA_INDEX_URL=https://repo.amd.com/rocm/whl-multi-arch/0 B
33ARG AMDGPU_FAMILY=device-all0 B
34ARG ROCM_VERSION=7.14.00 B
35ARG PYTORCH_ROCM_ARCH=gfx1010;gfx1011;gfx1012;gfx1030;gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx9500 B
36ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_SOURCE_DIR=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel ROCM_BIN=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin0 B
37ENV CMAKE_PREFIX_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/cmake HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode ROCM_SYSDEPS_INCLUDE=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/include0 B
38RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all1009.99 MB
39RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all6.81 GB
40COPY pytorch /workspace/pytorch # buildkit130.79 MB
41RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all176 B
42RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all1.43 GB
43RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all92 B
44RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all217 B
45RUN |13 PYTHON_VERSION=3.12 ROCM_VERSION=7.14.0 AMDGPU_FAMILY=device-all135.05 KB
46WORKDIR /0 B

Command

 ARG RELEASE