runpod/rocm:1.4.0-rc.167-rocm644-ubuntu2404-py312-pytorch260

Manifest digest

sha256:cc57280d72c59cde65d488ea45e702147896353845860d60fa91799494cedf61

OS/ARCH

linux/amd64

Compressed size

23.25 GB

Last pushed

14 days by runpod

Type

Image

Manifest digest

sha256:cc57280d72c59cde65d488ea45e702147896353845860d60fa91799494cedf61

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=24.040 B

5ADD file ... in / 28.35 MB

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

7ARG UBUNTU_VERSION0 B

8ENV DEBIAN_FRONTEND=noninteractive0 B

9ARG PYTORCH_ROCM_ARCH0 B

10ENV PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1150;gfx1151;gfx942;gfx1200;gfx12010 B

11COPY file:b2a090cb02b5615fc2d1036ea77428cbb70253c6a8b8614f78fab85c2ba2dfb5 in install_base.sh 1.84 KB

12|1 UBUNTU_VERSION=24.04 /bin/sh -c bash326.49 MB

13ARG LLVMDEV0 B

14ARG CLANG_VERSION0 B

15COPY file:579317ceda1019f6cd72be303e1ec7c4c2486dbb81a333e2747d779070ed6794 in install_clang.sh 861 B

16|3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=24.04 /bin/sh106 B

17COPY file:9f0cc865cc521cff044d880f1cb248b7f8e1ed60e8871fd662be075c2c6381b0 in install_user.sh 702 B

18|3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=24.04 /bin/sh1.72 KB

19ARG ANACONDA_PYTHON_VERSION0 B

20ARG CONDA_CMAKE0 B

21ARG BUILD_ENVIRONMENT0 B

22ENV ANACONDA_PYTHON_VERSION=3.120 B

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

24COPY file:91a5479240e04cc70fb36e504afe55d2b2f7e21dd29d31fdf52bbd0f04049a2e in /opt/conda/requirements-ci.txt 3.63 KB

25COPY file:4f2030250f761e6ac8f87880f43562c7195f8c965a62221ef19c1546241d8439 in install_conda.sh 1.91 KB

26COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 722 B

27|5 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes LLVMDEV=2.72 GB

28ARG GCC_VERSION0 B

29COPY file:f0e99070d8114620f665268ab11f181c0153236fb98685afc2502a7f1e978b2c in install_gcc.sh 378 B

30|6 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=104 B

31ARG PROTOBUF0 B

32COPY file:638eb6acc2fc2ff495707449113f05ffc78b46b8a4bb23639e30d89724372a1f in install_protobuf.sh 502 B

33|7 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=92.5 MB

34|7 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=109 B

35ENV INSTALLED_PROTOBUF=yes0 B

36ARG DB0 B

37COPY file:e2f252dfb0c5021405f1818c90ddc6a8fd573505ed7427bfbdd51167d3e5dda7 in install_db.sh 487 B

38|8 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes347 B

39|8 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes103 B

40ENV INSTALLED_DB=yes0 B

41ARG VISION0 B

42COPY multi:ecc701bddaf12d424d5629cf1acf0828dd4080da63bc673d1ddf7460edfab7e2 in ./ 1.84 KB

43|9 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes312.5 MB

44|9 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes164 B

45ENV INSTALLED_VISION=yes0 B

46ARG ROCM_VERSION0 B

47COPY file:97b22d5b61371bd4f097b3f607f7d0b6ded9cb3f757b0dc23b7c2375782f4ef6 in install_rocm.sh 1.65 KB

48|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes7.74 GB

49|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes105 B

50COPY file:3f84c25575a5ec8d17900dba4ad7cc244f7c9d6da474557f340ec501f674aa0c in install_rocm_magma.sh 1.23 KB

51|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes741.42 MB

52|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes111 B

53ADD file ... in install_miopen.sh 1.52 KB

54|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes107 B

55ENV ROCM_PATH=/opt/rocm0 B

56

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

0 B

57

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

0 B

58

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

0 B

59

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

0 B

60

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.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

61ENV MAGMA_HOME=/opt/rocm/magma0 B

62ENV LANG=C.UTF-80 B

63ENV LC_ALL=C.UTF-80 B

64COPY file:78e4a3281ca763191f3b77613d800ded4b7b00b181cdeec0200f96c67a3613ba in install_amdsmi.sh 167 B

65|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes2.84 MB

66|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes107 B

67ARG CMAKE_VERSION0 B

68COPY file:5201a03ab45964625e825e856290df6dfb1826d702a824e47020a4d65c575468 in install_cmake.sh 563 B

69|11 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes41.21 MB

70|11 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes106 B

71ARG NINJA_VERSION0 B

72COPY file:f562ac7c4f4bdd63c08a52ddfc7e1180d5ee49530c1a18f726242695cfae1868 in install_ninja.sh 294 B

73|12 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes86.23 KB

74|12 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes104 B

75ARG TRITON0 B

76COPY file:4278720cb36f67567d70b5828bbe9eb47e3b03c9ae7a67fa254f9478559cd993 in install_triton.sh 1.37 KB

77COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 722 B

78COPY file:52c5a134f1c7d9a4a4fc7e6118231651125f3222561cce11c3a82c2b6603da79 in triton.txt 138 B

79COPY file:317afa136ef0c0a5d80d6702afbfd60a0d37bb2987570d2514318ccc8051e223 in triton_version.txt 118 B

80|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes4.1 GB

81|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes172 B

82COPY file:62caab7ea7f18fb87560e07354fbefedb8b0eab2c4e3ed0bfa080fb1a2394d69 in install_cache.sh 1.88 KB

83

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.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

84|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes15.24 MB

85ARG BUILD_ENVIRONMENT0 B

86ENV BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.4-py3.120 B

87COPY dir:4ccc9e9d18635493a3825294a70103eaa912bdc81d80351d4dcd16f0a2e3d2b1 in /opt/llvm 51.64 MB

88USER jenkins0 B

89CMD ["bash"]0 B

90ARG FRAME_NAME0 B

91ARG OS_CHOICE0 B

92ARG PY_VER0 B

93ARG PY_EXTRA_PK0 B

94ARG MAX_JOBS=40 B

95ENV MAX_JOBS=320 B

96USER root0 B

97COPY dir:f11d39cc79ba159388d5cfdc38e531b54c58144bceacccb785ffd9ffe15f69e7 in /var/lib/jenkins 6.93 KB

98COPY dir:b5f1dc67ec8c94498de01f06391a3f120a5571874ab2e4c2b58e28e80e7bc59d in /var/lib/jenkins/pytorch 3.51 GB

99WORKDIR /var/lib/jenkins0 B

100|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu24.04 PY_EXTRA_PK= PY_VER=py3.120 B

101|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu24.04 PY_EXTRA_PK= PY_VER=py3.123.07 GB

102ENV ROCM_PATH=/opt/rocm0 B

103ENV MPI_HOME=/opt/ompi0 B

104ENV UCX_HOME=/opt/ucx0 B

105

ENV PATH=/opt/ompi/bin:/opt/ucx/bin:/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.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

106ENV LD_LIBRARY_PATH=/opt/ompi/lib:/opt/rocm/lib:/usr/local/lib:0 B

107ENV CI=10 B

108ENV PYTORCH_TEST_WITH_ROCM=10 B

109ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B

110ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B

111ARG RP_SKIP_PYTHON=10 B

112SHELL [/bin/bash -o pipefail -c]0 B

113ENV SHELL=/bin/bash0 B

114ENV PYTHONUNBUFFERED=True0 B

115ENV DEBIAN_FRONTEND=noninteractive0 B

116ENV RP_WORKSPACE=/workspace0 B

117ENV HF_HOME=/workspace/.cache/huggingface/0 B

118ENV VIRTUALENV_OVERRIDE_APP_DATA=/workspace/.cache/virtualenv/0 B

119ENV PIP_CACHE_DIR=/workspace/.cache/pip/0 B

120ENV UV_CACHE_DIR=/workspace/.cache/uv/0 B

121ENV HF_HUB_ENABLE_HF_TRANSFER=10 B

122ENV HF_XET_HIGH_PERFORMANCE=10 B

123ENV PIP_BREAK_SYSTEM_PACKAGES=10 B

124ENV PIP_ROOT_USER_ACTION=ignore0 B

125ENV TZ=Etc/UTC0 B

126WORKDIR /0 B

127RUN |1 RP_SKIP_PYTHON=1 /bin/bash -o153 B

128RUN |1 RP_SKIP_PYTHON=1 /bin/bash -o417.05 MB

129RUN |1 RP_SKIP_PYTHON=1 /bin/bash -o32 B

130RUN |1 RP_SKIP_PYTHON=1 /bin/bash -o32 B

131ARG GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d4540 B

132ARG GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a760 B

133ARG GET_PIP_39_SHA256=95c0ae79ccf9ac1e47e4187f49081d6ae1f45997afd4549e14f37337cbcfd7660 B

134

RUN |4 RP_SKIP_PYTHON=1 GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d454 GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a76

32 B

135

RUN |4 RP_SKIP_PYTHON=1 GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d454 GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a76

32 B

136

RUN |4 RP_SKIP_PYTHON=1 GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d454 GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a76

32 B

137COPY /uv /uvx /bin/ #21.56 MB

138COPY requirements.txt /requirements.txt # buildkit378 B

139COPY scrub-stale-metadata.py /tmp/scrub-stale-metadata.py # buildkit1.29 KB

140

RUN |4 RP_SKIP_PYTHON=1 GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d454 GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a76

109.39 MB

141COPY nginx.conf /etc/nginx/nginx.conf # buildkit986 B

142COPY snippets /etc/nginx/snippets # buildkit558 B

143COPY readme.html /usr/share/nginx/html/readme.html # buildkit831 B

144COPY README.md /usr/share/nginx/html/README.md # buildkit1.1 KB

145COPY --chmod=755 start.sh / #1.25 KB

146COPY runpod.txt /etc/runpod.txt # buildkit234 B

147

RUN |4 RP_SKIP_PYTHON=1 GET_PIP_COMMIT=0d4515b4d114f9cd820446a4bd4bd0986393d454 GET_PIP_SHA256=66904bccb878e363db6236ea900e6935e507dcb887e9f178f6212edfe7f46a76

1.61 KB

148CMD ["/start.sh"]0 B

Command

 ARG RELEASE