srnbckr/slurm-rocm:latest

Manifest digest

sha256:aae51f90a950a3cb17445955348a5cb65ce52ac423ba678ae9fd686f9bf44ad6

OS/ARCH

linux/amd64

Compressed size

27.77 GB

Last pushed

12 months by srnbckr

Type

Image

Manifest digest

sha256:aae51f90a950a3cb17445955348a5cb65ce52ac423ba678ae9fd686f9bf44ad6

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 / 29.03 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;gfx940;gfx941;gfx942;gfx1200;gfx12010 B

11COPY file:92bc590372c306f3daa67db86d673b9035c09b5f87f0d34625855bea00a6a7bd in install_base.sh 2.04 KB

12|1 UBUNTU_VERSION=22.04 /bin/sh -c bash324.43 MB

13ARG LLVMDEV0 B

14ARG CLANG_VERSION0 B

15COPY file:ea4470f09a1a3c3e7b5e68367ef79048b788247d3169e9da385ac1a9afb3216a in install_clang.sh 803 B

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

17COPY file:68c4e8562a92679b6952c627851f17ce90159361e6fc1deec5dab1746a5fc809 in install_user.sh 703 B

18|3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=22.04 /bin/sh1.21 KB

19ARG ANACONDA_PYTHON_VERSION0 B

20ARG CONDA_CMAKE0 B

21ENV ANACONDA_PYTHON_VERSION=3.100 B

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

23COPY file:642a6d9430901a23a9d611c911c53eba2196593b07f347a8e5abad7a868e3f74 in /opt/conda/requirements-ci.txt 3.27 KB

24COPY file:e8af78b500d54f26fa4bf7d1864835b7e3a60436c96f8e4f2101214895534b7a in install_conda.sh 2.01 KB

25COPY file:c5302e49e297c6cd3705786f39312b29dc4a58c734f1c1a397ca2038754c6af5 in common_utils.sh 724 B

26|4 CLANG_VERSION= CONDA_CMAKE=yes LLVMDEV= UBUNTU_VERSION=22.042.43 GB

27ARG GCC_VERSION0 B

28COPY file:a6be3fca5763d553ed3491dd421b101b6a4ea1af953b169ac5c98b0e15ecbe62 in install_gcc.sh 380 B

29|5 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV=104 B

30ARG PROTOBUF0 B

31COPY file:117aa4dec9c82c451f03ba4c0bd2962cd06eca87c9c99990af1049cfb39c524c in install_protobuf.sh 503 B

32|6 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV=96.43 MB

33|6 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION= LLVMDEV=109 B

34ENV INSTALLED_PROTOBUF=yes0 B

35ARG DB0 B

36COPY file:66fd685a986e705c33ce901c8601ca2672443d3f9b70c37b72f480ca31613a6b in install_db.sh 490 B

37|7 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=352 B

38|7 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=103 B

39ENV INSTALLED_DB=yes0 B

40ARG VISION0 B

41COPY multi:6e571e37fa8fc7c93beea9fa3343c007e445aca707b9ecdaa4740f12f475a571 in ./ 1.84 KB

42|8 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=227.06 MB

43|8 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=160 B

44ENV INSTALLED_VISION=yes0 B

45ARG ROCM_VERSION0 B

46COPY file:2c8ff5dbee0197b061b98f1f279ea1a17c1950b3c66298aef62640d94744c6e8 in install_rocm.sh 1.68 KB

47|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=8.02 GB

48|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=105 B

49COPY file:d98119f5af1b9faf43e5508c7dff2c997ec9066336c0b7117755c698c92006fb in install_rocm_magma.sh 841 B

50|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=725.7 MB

51|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=111 B

52ENV ROCM_PATH=/opt/rocm0 B

53

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

54

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

55

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

56

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

57

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

58ENV MAGMA_HOME=/opt/rocm/magma0 B

59ENV LANG=C.UTF-80 B

60ENV LC_ALL=C.UTF-80 B

61COPY file:318c3352c597ff6231f957520a2f1d120cd17d46642c3aed1a681a0e4aabd7fd in install_amdsmi.sh 166 B

62|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=2.56 MB

63|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=107 B

64ARG CMAKE_VERSION0 B

65COPY file:81e429df46fa0455cf421853efc016e51afc523e70e51c6582fa7b29fe87e4bb in install_cmake.sh 565 B

66|10 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes41.21 MB

67|10 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes106 B

68ARG NINJA_VERSION0 B

69COPY file:00cb9f7fa5c380835603121c64a21edd9ecfa3bac0af0ec60f552a7669c314e7 in install_ninja.sh 295 B

70|11 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes86.22 KB

71|11 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes106 B

72ARG TRITON0 B

73COPY file:c805c90c8b8fdc86478eb71bbfeeeb3ad1a3f54ea81b2da45696e171599b8cb5 in install_triton.sh 1.18 KB

74COPY file:c5302e49e297c6cd3705786f39312b29dc4a58c734f1c1a397ca2038754c6af5 in common_utils.sh 724 B

75COPY file:caf1492a188d87bdd2bc45b1746f4e4332aa9cc869d43eda0529b663e203d923 in triton-rocm.txt 140 B

76COPY file:c2b21f03dcf0f51754d4e6b572b5498c8ca5fbeb1faa74582bb304fb2222198c in triton_version.txt 120 B

77|12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes1.8 GB

78|12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes177 B

79COPY file:0838d251e98cd169718c8b92dd9587a248266bb3b15480aa1b69fea3b91181b5 in install_cache.sh 1.8 KB

80

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

81|12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes15.24 MB

82ARG BUILD_ENVIRONMENT0 B

83ENV BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.3-py3.100 B

84USER jenkins0 B

85CMD ["bash"]0 B

86ARG FRAME_NAME0 B

87ARG OS_CHOICE0 B

88ARG PY_VER0 B

89ARG PY_EXTRA_PK0 B

90ARG MAX_JOBS=40 B

91ENV MAX_JOBS=320 B

92USER root0 B

93COPY dir:93ff6081f0f276da50e74944615e439739ab91faf470f0e6230e2cf33bafecd5 in /var/lib/jenkins 6.85 KB

94COPY dir:496f0dce1c7858ab8188d7491b98d179d7603ce9bddf5ff65f0f294198d14721 in /var/lib/jenkins/pytorch 2.99 GB

95WORKDIR /var/lib/jenkins0 B

96|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_EXTRA_PK= PY_VER=py3.100 B

97|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_EXTRA_PK= PY_VER=py3.105 GB

98ENV ROCM_PATH=/opt/rocm0 B

99ENV MPI_HOME=/opt/ompi0 B

100ENV UCX_HOME=/opt/ucx0 B

101

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

0 B

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

103ENV CI=10 B

104ENV PYTORCH_TEST_WITH_ROCM=10 B

105ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B

106ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B

107RUN /bin/sh -c apt update118.34 MB

108RUN /bin/sh -c python3 -m4.77 MB

109RUN /bin/sh -c pip install52.68 MB

110RUN /bin/sh -c pip install31.58 MB

111RUN /bin/sh -c pip install14.3 MB

112RUN /bin/sh -c pip uninstall593 B

113RUN /bin/sh -c apt --fix-broken394 B

114RUN /bin/sh -c apt install1.17 MB

115RUN /bin/sh -c apt install1.64 MB

116ENV LLVM_SYMBOLIZER_PATH=/opt/rocm/llvm/bin/llvm-symbolizer0 B

117

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

0 B

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

119ENV PYTORCH_ROCM_ARCH=gfx9420 B

120ARG MAX_JOBS=2560 B

121RUN |1 MAX_JOBS=256 /bin/sh -c402.44 MB

122RUN |1 MAX_JOBS=256 /bin/sh -c282.98 MB

123ARG PYTORCH_BRANCH=63743320 B

124RUN |2 MAX_JOBS=256 PYTORCH_BRANCH=6374332 /bin/sh723.45 MB

125RUN |2 MAX_JOBS=256 PYTORCH_BRANCH=6374332 /bin/sh538.16 MB

126ARG GPU_ARCHS=gfx9420 B

127RUN |3 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx94243.02 MB

128RUN |3 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx942483.07 MB

129ENV NVTE_USE_HIPBLASLT=10 B

130ENV NVTE_FRAMEWORK=pytorch0 B

131ENV NVTE_ROCM_ARCH=gfx9420 B

132ENV NVTE_USE_CAST_TRANSPOSE_TRITON=00 B

133ENV NVTE_CK_USES_BWD_V3=10 B

134ENV NVTE_CK_V3_BF16_CVT=20 B

135RUN |3 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx9422.42 GB

136ARG HIPBLASLT_BRANCH=258a21620 B

137RUN |4 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx942587.82 MB

138ENV HIP_FORCE_DEV_KERNARG=10 B

139RUN |4 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx942709.54 KB

140RUN |4 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx942191.43 MB

141RUN |4 MAX_JOBS=256 PYTORCH_BRANCH=6374332 GPU_ARCHS=gfx942933.9 KB

142WORKDIR /workspace/99 B

143CMD ["/usr/bin/bash"]0 B

144/bin/bash15.01 MB

145/bin/bash39.63 KB

146/bin/bash39.77 KB

147/bin/bash32.3 KB

148/bin/bash26.88 MB

149/bin/bash125.5 MB

150/bin/sh64.14 MB

151SHELL [bash -c]0 B

152ARG DEBIAN_FRONTEND=noninteractive0 B

153ARG SLURM_VERSION=24.05.70 B

154ENV SLURM_VERSION=24.05.70 B

155

LABEL [email protected] org.opencontainers.image.documentation=https://slurm.schedmd.com/documentation.html org.opencontainers.image.license=GPL-2.0-or-later WITH

0 B

156LABEL vendor=SchedMD LLC. version=24.05.7 release=https://github.com/SlinkyProject/containers0 B

157USER root0 B

158WORKDIR /tmp/32 B

159ARG SLURM_USER=slurm0 B

160ARG SLURM_USER_UID=4010 B

161ARG SLURM_USER_GID=4010 B

162RUN |5 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm1.38 KB

163COPY /tmp/*.deb /tmp/ # buildkit28.66 MB

164RUN |5 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm1.45 MB

165ENV KUBECTL_VERSION=1.320 B

166RUN |5 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm19.93 MB

167ARG DEBUG=00 B

168RUN |6 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm32 B

169SHELL [bash -c]0 B

170ARG DEBIAN_FRONTEND=noninteractive0 B

171LABEL org.opencontainers.image.title=Slurm Auth/Cred Server org.opencontainers.image.description=sackd0 B

172LABEL name=Slurm Auth/Cred Server summary=sackd0 B

173USER root0 B

174WORKDIR /tmp/32 B

175RUN |6 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm7.94 MB

176RUN |6 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm332 B

177RUN |6 DEBIAN_FRONTEND=noninteractive SLURM_VERSION=24.05.7 SLURM_USER=slurm1.34 KB

178ENTRYPOINT ["tini" "-g" "--" "sackd.sh"]0 B

Command

 ARG RELEASE