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

Manifest digest

sha256:36af6ab64ecf7cfa6e1d8b60717165273dd185468e1197cef333eb7f513f2a39

OS/ARCH

linux/amd64

Compressed size

10.91 GB

Last pushed

5 days by rocm

Type

Image

Manifest digest

sha256:36af6ab64ecf7cfa6e1d8b60717165273dd185468e1197cef333eb7f513f2a39

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

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

4ADD file ... in / 29.21 MB

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

6ARG VERSION=10.1.00 B

7ARG TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/0 B

8ARG TORCH_SPECS=torch[device-all]==2.12.0+rocm10.1.00 B

9ARG TORCHVISION_SPECS=torchvision[device-all]==0.27.0+rocm10.1.00 B

10ARG TORCHAUDIO_SPECS=torchaudio==2.11.0+rocm10.1.00 B

11ARG APEX_SPECS=apex==1.12.0+rocm10.1.00 B

12LABEL org.opencontainers.image.title=ROCm PyTorch (rocm-ai-docker-stack) org.opencontainers.image.description=Multi-arch0 B

13COPY staging/install_rocm_deps.sh /tmp/install_rocm_deps.sh # buildkit1.31 KB

14

RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.12.0+rocm10.1.0

134.14 MB

15

RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.12.0+rocm10.1.0

195 B

16

RUN |6 VERSION=10.1.0 TORCH_INDEX_URL=https://d1kq5rufa68c0p.cloudfront.net/rocm/whl-next/ TORCH_SPECS=torch[device-all]==2.12.0+rocm10.1.0

10.75 GB

17COPY --chmod=0755 scripts/smoke_pytorch.sh /usr/local/bin/ #707 B

18COPY --chmod=0755 scripts/setup_pytorch_env.sh /usr/local/bin/setup_pytorch_env #10.54 KB

19ENV PYTHONUNBUFFERED=10 B

Command

 ARG RELEASE