rocm/amd-mlperf:bert_training_5.0

Manifest digest

sha256:3404043497e53775715073555997dc5ba99903954fccc34ef314b424fe432cf6

OS/ARCH

linux/amd64

Compressed size

34.73 GB

Last pushed

about 1 year by rocmdev

Type

Image

Manifest digest

sha256:3404043497e53775715073555997dc5ba99903954fccc34ef314b424fe432cf6

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;gfx942;gfx1200;gfx12010 B

11COPY file:b2a090cb02b5615fc2d1036ea77428cbb70253c6a8b8614f78fab85c2ba2dfb5 in install_base.sh 1.84 KB

12|1 UBUNTU_VERSION=22.04 /bin/sh -c bash298.77 MB

13ARG LLVMDEV0 B

14ARG CLANG_VERSION0 B

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

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

17COPY file:9f0cc865cc521cff044d880f1cb248b7f8e1ed60e8871fd662be075c2c6381b0 in install_user.sh 704 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:5bb2bf9f600c791c8add13c6b7fa012e87171d16f0ff9cc6e36b5fa85ef534e8 in /opt/conda/requirements-ci.txt 3.71 KB

24COPY file:8fe8e351aea1e010b0117a4254bd72e4a2bfe629963ab95a4c78b670ebdba0e0 in install_conda.sh 1.86 KB

25COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 724 B

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

27ARG GCC_VERSION0 B

28COPY file:f0e99070d8114620f665268ab11f181c0153236fb98685afc2502a7f1e978b2c in install_gcc.sh 380 B

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

30ARG PROTOBUF0 B

31COPY file:638eb6acc2fc2ff495707449113f05ffc78b46b8a4bb23639e30d89724372a1f in install_protobuf.sh 505 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:e2f252dfb0c5021405f1818c90ddc6a8fd573505ed7427bfbdd51167d3e5dda7 in install_db.sh 490 B

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

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

39ENV INSTALLED_DB=yes0 B

40ARG VISION0 B

41COPY multi:ecc701bddaf12d424d5629cf1acf0828dd4080da63bc673d1ddf7460edfab7e2 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=161 B

44ENV INSTALLED_VISION=yes0 B

45ARG ROCM_VERSION0 B

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

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

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

49COPY file:dc007ec718680ca3fa2f00ec07dcee0d2f25ef694073fdb79ac48f76140b21d5 in install_rocm_magma.sh 1.09 KB

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

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

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

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

54ENV ROCM_PATH=/opt/rocm0 B

55

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

56

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

57

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

58

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

59

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

60ENV MAGMA_HOME=/opt/rocm/magma0 B

61ENV LANG=C.UTF-80 B

62ENV LC_ALL=C.UTF-80 B

63COPY file:78e4a3281ca763191f3b77613d800ded4b7b00b181cdeec0200f96c67a3613ba in install_amdsmi.sh 166 B

64|9 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=2.72 MB

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

66ARG CMAKE_VERSION0 B

67COPY file:5201a03ab45964625e825e856290df6dfb1826d702a824e47020a4d65c575468 in install_cmake.sh 565 B

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

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

70ARG NINJA_VERSION0 B

71COPY file:f562ac7c4f4bdd63c08a52ddfc7e1180d5ee49530c1a18f726242695cfae1868 in install_ninja.sh 296 B

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

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

74ARG TRITON0 B

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

76COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 724 B

77COPY file:77ff88eac004d39867271f88970329615a9fc3b28797c7bf4cc1caf5ede2c2f0 in triton.txt 137 B

78COPY file:317afa136ef0c0a5d80d6702afbfd60a0d37bb2987570d2514318ccc8051e223 in triton_version.txt 120 B

79|12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes3.95 GB

80|12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes DB=yes172 B

81COPY file:62caab7ea7f18fb87560e07354fbefedb8b0eab2c4e3ed0bfa080fb1a2394d69 in install_cache.sh 1.89 KB

82

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

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

84ARG BUILD_ENVIRONMENT0 B

85ENV BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm6.4-py3.100 B

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

87USER jenkins0 B

88CMD ["bash"]0 B

89ARG FRAME_NAME0 B

90ARG OS_CHOICE0 B

91ARG PY_VER0 B

92ARG PY_EXTRA_PK0 B

93ARG MAX_JOBS=40 B

94ENV MAX_JOBS=320 B

95USER root0 B

96COPY dir:ace6263b7df865e97efb5c7b8542262b8209354dcb40b3ac2ea64cd79ddc5c5a in /var/lib/jenkins 6.93 KB

97COPY dir:1bbc465e1cc099139bbf1d9ab106f09a854df2e16c0080253ab6c7cf8a9eb34a in /var/lib/jenkins/pytorch 3.29 GB

98WORKDIR /var/lib/jenkins0 B

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

100|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu22.04 PY_EXTRA_PK= PY_VER=py3.102.77 GB

101ENV ROCM_PATH=/opt/rocm0 B

102ENV MPI_HOME=/opt/ompi0 B

103ENV UCX_HOME=/opt/ucx0 B

104

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

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

106ENV CI=10 B

107ENV PYTORCH_TEST_WITH_ROCM=10 B

108ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B

109ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B

110WORKDIR /workspace98 B

111WORKDIR /workspace/deps119 B

112COPY file:420a845c0aa0f1e8a49116129d0cf6707ad65804c13ce020f01ccb26bd35ec71 in requirements.txt 356 B

113/bin/sh -c pip3 install -r1.37 GB

114COPY dir:f606b0195a1c648adfe8eef2b2c6d836e0ed50cede9fbd544a182167033374cf in . 14.23 KB

115/bin/sh -c cd mhalib &&425.44 KB

116/bin/sh -c git clone https://github.com/ROCm/flash-attention/299.34 MB

117/bin/sh -c git clone --recursive1.02 GB

118/bin/sh -c git clone https://github.com/ROCm/hipBLASLt.git11.4 GB

119WORKDIR /workspace/code118 B

120COPY dir:d23b0dca8c8c4afdb53d70636a5e345907def724109340aed5cae836b6cb5cd3 in . 500.36 KB

Command

 ARG RELEASE