sunway513/pytorch:rocm6.3_ubuntu22.04_py3.10_pytorch_release_2.5.1_flashattention

Manifest digest

sha256:ff16a49dc403375b0123b14234128ce5620ae86c0d3c5168910792eca49c3013

OS/ARCH

linux/amd64

Compressed size

24.7 GB

Last pushed

about 1 year by sunway513

Type

Image

Manifest digest

sha256:ff16a49dc403375b0123b14234128ce5620ae86c0d3c5168910792eca49c3013

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_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:f116edf07e857a09ef18f25334340f809338acb41944ed9d3ece0429b86474b6 in install_base.sh 1.84 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:c6304d808aa099de4888e640fb5596877c462b857b7c670e8d091c4165ccf0a4 in install_user.sh 544 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:44ba7b4d2766003cdee0acd82b8a5908eff91f9dfc8d473c419abc9ef4930596 in /opt/conda/requirements-ci.txt 3.41 KB

24COPY file:23c2adc1db149ce62b510d41d5edc9ee9b5e162ae6a2c5e3006fbd33c05fc988 in install_conda.sh 1.97 KB

25COPY file:c5302e49e297c6cd3705786f39312b29dc4a58c734f1c1a397ca2038754c6af5 in common_utils.sh 723 B

26|4 CLANG_VERSION= CONDA_CMAKE=yes LLVMDEV= UBUNTU_VERSION=22.042.48 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 504 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 488 B

37|7 CLANG_VERSION= CONDA_CMAKE=yes DB=yes GCC_VERSION=354 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:0f870960e7bcc404bc3ae41dc4fad1c080cd8f4190bc4ec2cab8ad02056d1fd3 in install_rocm.sh 1.67 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:07a429a59c23abf610a1d5aac73f7550a43004443a152a4b4e51868b149bb189 in install_rocm_magma.sh 974 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

52ADD file ... in install_miopen.sh 1.6 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:318c3352c597ff6231f957520a2f1d120cd17d46642c3aed1a681a0e4aabd7fd in install_amdsmi.sh 166 B

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

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

66ARG CMAKE_VERSION0 B

67COPY file:81e429df46fa0455cf421853efc016e51afc523e70e51c6582fa7b29fe87e4bb in install_cmake.sh 564 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:00cb9f7fa5c380835603121c64a21edd9ecfa3bac0af0ec60f552a7669c314e7 in install_ninja.sh 295 B

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

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

74ARG TRITON0 B

75COPY file:a383ab81661d417d65c7ed3e91f442b1b596d212acdc59415f4c2aa66bbf01a5 in install_triton.sh 1.33 KB

76COPY file:c5302e49e297c6cd3705786f39312b29dc4a58c734f1c1a397ca2038754c6af5 in common_utils.sh 723 B

77COPY file:6e3f0601f8c2abf2f1874b02e9a12b622f4997deeca02e595d03c0407afdc67c in triton.txt 138 B

78COPY file:588f07dedc94268a8d976c0f6a18384b941ec93c94eca37300613a97d7893222 in triton_version.txt 119 B

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

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

81COPY file:0838d251e98cd169718c8b92dd9587a248266bb3b15480aa1b69fea3b91181b5 in install_cache.sh 1.8 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.3-py3.100 B

86COPY dir:c3ac8a3fe71e0f2d818f1fe0fa1ee6e7cfbff0675bd2436ddcd87ea9721a4c9f 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:e3019adac44444e1eae25c38613b9bde26d6a5976cd036fede8c66b8db6e2b65 in /var/lib/jenkins 6.85 KB

97COPY dir:ddda3a6f03901dec22356196152128b17f06422f8337a16ebcff08de7492c3dc in /var/lib/jenkins/pytorch 3.03 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.105.79 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

110RUN /bin/sh -c git clone392.13 MB

Command

 ARG RELEASE