genesu100/utc_ci:30037742212

Index digest

sha256:8838145eb5b6847e7974daff8d06725d6458cec8fa106501d2518bdbc6412117

OS/ARCH

linux/amd64

Compressed size

10.29 GB

Last pushed

3 months by genesu100

Type

Image

Manifest digest

sha256:08f27db20076a4978b106249b3b430d8350774b5681401881a66aa07469d5fdb

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.version=24.040 B

4ADD file ... in / 28.36 MB

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

6WORKDIR /workspace/99 B

7ENV MAX_JOBS=1280 B

8ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

9ENV ROCM_AMDGPU_TARGETS=gfx942,gfx9500 B

10ENV ROCPROFILER_QUEUE_INTERPOSITION=00 B

11ENV DEBUG_HIP_DYNAMIC_QUEUES=00 B

12ENV DEBIAN_FRONTEND=noninteractive0 B

13RUN /bin/sh -c apt update276.71 MB

14RUN /bin/sh -c python -m4 MB

15ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

16RUN /bin/sh -c pip install32.16 MB

17ENV THE_ROCK_VERSION=7.15.0a202607210 B

18RUN /bin/sh -c python -m3.24 GB

19RUN /bin/sh -c rocm-sdk init3.24 GB

20ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

21ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

22ENV HIP_PLATFORM=amd0 B

23ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

24ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B

25ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

26ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

27ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B

28

ENV PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

29

ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib

0 B

30

ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64

0 B

31ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

32ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B

33ARG TF_REPO=https://github.com/ROCm/tensorflow-upstream.git0 B

34ARG TF_BRANCH=upstream-v2.21.00 B

35ARG BAZELISK_VERSION=v1.25.00 B

36RUN |3 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.03 MB

37RUN |3 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.0264.58 MB

38ARG JAX_VERSION=0.10.20 B

39ENV JAX_VERSION=0.10.20 B

40ENV JAX_PJRT_VERSION=0.10.2+rocm7.15.0a202607210 B

41ENV JAX_PLUGIN_VERSION=0.10.2+rocm7.15.0a202607210 B

42RUN |4 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.0287 MB

43ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a10 B

44RUN |5 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.0364.73 MB

45ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B

46ENV NVTE_USE_HIPBLASLT=10 B

47ENV GPU_MAX_HW_QUEUES=20 B

48ENV HIP_FORCE_DEV_KERNARG=10 B

49ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B

50ENV NVTE_FUSED_ATTN=10 B

51ENV NCCL_DEBUG=VERSION0 B

52ENV NVTE_CK_USES_BWD_V3=10 B

53ENV NVTE_CK_USES_FWD_V3=10 B

54ENV NVTE_CK_IS_V3_ATOMIC_FP32=10 B

55ENV NVTE_CK_HOW_V3_BF16_CVT=20 B

56ENV XLA_PYTHON_CLIENT_MEM_FRACTION=.90 B

57

ENV XLA_FLAGS=--xla_gpu_memory_limit_slop_factor=95 --xla_gpu_reduce_scatter_combine_threshold_bytes=8589934592 --xla_gpu_enable_latency_hiding_scheduler=True --xla_gpu_all_gather_combine_threshold_bytes=8589934592

0 B

58WORKDIR /workspace32 B

59ARG MAXTEXT_REPO=https://github.com/ROCm/maxtext.git0 B

60ARG MAXTEXT_BRANCH=release/v26.50 B

61RUN |7 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.01.48 GB

62ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B

63ARG PRIMUS_BRANCH=main0 B

64RUN |9 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.01005.35 MB

65RUN |9 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.026.29 MB

66ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B

67RUN |10 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.02.64 MB

68RUN |10 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.0225 B

69ARG UCX_VERSION=1.18.00 B

70RUN |11 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.086.43 MB

71ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B

72ARG MPI_VERSION=4.1.60 B

73RUN |12 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.03.84 MB

74RUN |12 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.0443 B

75RUN |12 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.032 B

76ARG GIT_COMMIT_TAG=ab27f584bfa08634b6870164351b35a4120eac2a0 B

77ARG DOCKERFILE_PATH=jax_training_the_rock/docker/Dockerfile0 B

78RUN |14 TF_REPO=https://github.com/ROCm/tensorflow-upstream.git TF_BRANCH=upstream-v2.21.0 BAZELISK_VERSION=v1.25.016.95 KB

79COPY jax_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit3.76 KB

Command

 ARG RELEASE