jeffdaily/pytorch:jammy-rocm-7.1-py3-20251031-base

Manifest digest

sha256:0f8f80d2a845a664ecdf9ad463f7a87cc175dea4ac521ae05205239d98e516bf

OS/ARCH

linux/amd64

Compressed size

17.65 GB

Last pushed

11 months by jeffdaily

Type

Image

Manifest digest

sha256:0f8f80d2a845a664ecdf9ad463f7a87cc175dea4ac521ae05205239d98e516bf

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 / 28.17 MB

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

7ARG UBUNTU_VERSION=22.040 B

8ENV DEBIAN_FRONTEND=noninteractive0 B

9ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx11000 B

10ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx11000 B

11COPY ./common/install_base.sh install_base.sh # buildkit1.52 KB

12RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 /bin/sh299.1 MB

13COPY ./common/install_user.sh install_user.sh # buildkit700 B

14RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 /bin/sh1.19 KB

15ARG KATEX=yes0 B

16COPY ./common/install_docs_reqs.sh install_docs_reqs.sh # buildkit481 B

17RUN |3 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes105.25 MB

18ARG ANACONDA_PYTHON_VERSION=3.100 B

19ARG BUILD_ENVIRONMENT=:pytorch-linux-jammy-rocm-n-py30 B

20ENV ANACONDA_PYTHON_VERSION=3.100 B

21ENV PATH=/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

22COPY requirements-ci.txt /opt/conda/requirements-ci.txt # buildkit4.27 KB

23COPY ./common/install_conda.sh install_conda.sh # buildkit1.71 KB

24COPY ./common/common_utils.sh common_utils.sh # buildkit718 B

25RUN |5 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes3.05 GB

26ARG GCC_VERSION=110 B

27COPY ./common/install_gcc.sh install_gcc.sh # buildkit378 B

28RUN |6 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes231.11 KB

29ARG CLANG_VERSION=0 B

30COPY ./common/install_clang.sh install_clang.sh # buildkit788 B

31RUN |7 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes106 B

32ARG VISION=yes0 B

33COPY ./common/install_vision.sh ./common/cache_vision_models.sh ./common/common_utils.sh ./1.47 KB

34RUN |8 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes436.93 MB

35RUN |8 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes163 B

36ENV INSTALLED_VISION=yes0 B

37ARG ROCM_VERSION=7.10 B

38RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes104 B

39COPY ./common/common_utils.sh common_utils.sh # buildkit718 B

40COPY ./ci_commit_pins/rocm-composable-kernel.txt ci_commit_pins/rocm-composable-kernel.txt # buildkit198 B

41COPY ./common/install_rocm.sh install_rocm.sh # buildkit2.52 KB

42RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes7.06 GB

43RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes135 B

44RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes104 B

45COPY ./common/install_rocm_magma.sh install_rocm_magma.sh # buildkit624 B

46RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes639.75 MB

47RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes111 B

48ADD ./common/install_miopen.sh install_miopen.sh # buildkit1.52 KB

49RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes107 B

50ENV ROCM_PATH=/opt/rocm0 B

51

ENV PATH=/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

52

ENV PATH=/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

53

ENV PATH=/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

54

ENV PATH=/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

55

ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

56ENV MAGMA_HOME=/opt/rocm/magma0 B

57ENV LANG=C.UTF-80 B

58ENV LC_ALL=C.UTF-80 B

59COPY ./common/install_amdsmi.sh install_amdsmi.sh # buildkit165 B

60RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes3.36 MB

61RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes107 B

62ARG UCX_COMMIT=29831d319e6be55cb8c768ca61de335c934ca39e0 B

63ARG UCC_COMMIT=9f4b242cbbd8b1462cbc732eb29316cdfa124b770 B

64ENV UCX_COMMIT=29831d319e6be55cb8c768ca61de335c934ca39e0 B

65ENV UCC_COMMIT=9f4b242cbbd8b1462cbc732eb29316cdfa124b770 B

66ENV UCX_HOME=/usr0 B

67ENV UCC_HOME=/usr0 B

68ADD ./common/install_ucc.sh install_ucc.sh # buildkit829 B

69RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes25.72 MB

70RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes104 B

71COPY ./common/install_openssl.sh install_openssl.sh # buildkit424 B

72ENV OPENSSL_ROOT_DIR=/opt/openssl0 B

73RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes18.41 MB

74ENV OPENSSL_DIR=/opt/openssl0 B

75ARG INDUCTOR_BENCHMARKS=0 B

76ARG ANACONDA_PYTHON_VERSION=3.100 B

77ENV ANACONDA_PYTHON_VERSION=3.100 B

78COPY ./common/install_inductor_benchmark_deps.sh install_inductor_benchmark_deps.sh # buildkit640 B

79COPY ./common/common_utils.sh common_utils.sh # buildkit718 B

80COPY ci_commit_pins/huggingface-requirements.txt huggingface-requirements.txt # buildkit148 B

81COPY ci_commit_pins/timm.txt timm.txt # buildkit135 B

82COPY ci_commit_pins/torchbench.txt torchbench.txt # buildkit140 B

83RUN |12 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes32 B

84RUN |12 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes223 B

85ARG NINJA_VERSION=1.9.00 B

86COPY ./common/install_ninja.sh install_ninja.sh # buildkit341 B

87RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes86.23 KB

88RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes106 B

89ARG TRITON=yes0 B

90COPY ./common/install_triton.sh install_triton.sh # buildkit1.63 KB

91COPY ./common/common_utils.sh common_utils.sh # buildkit718 B

92COPY ci_commit_pins/triton.txt triton.txt # buildkit139 B

93COPY triton_version.txt triton_version.txt # buildkit120 B

94RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes5.79 GB

95RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes176 B

96COPY ./common/install_cache.sh install_cache.sh # buildkit1.85 KB

97

ENV PATH=/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

98RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes188.42 MB

99COPY ./common/install_openmpi.sh install_openmpi.sh # buildkit300 B

100RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes32 B

101RUN |14 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100 KATEX=yes108 B

102ARG BUILD_ENVIRONMENT=:pytorch-linux-jammy-rocm-n-py30 B

103ENV BUILD_ENVIRONMENT=:pytorch-linux-jammy-rocm-n-py30 B

104COPY /opt/llvm /opt/llvm # buildkit51.64 MB

105USER jenkins0 B

106CMD ["bash"]0 B

Command

 ARG RELEASE