Sign inSign up

genesu100/utc_ci:30106425102

Index digest

sha256:10762a6a92c910d051c89df2ad18d4cf099c2141546e8f5db4d65173e80991c2

OS/ARCH

linux/amd64

Compressed size

10.29 GB

Last pushed

3 months by genesu100

Type

Image

Manifest digest

sha256:e118010c240efd65413b9468d622ba4d2796262f8e8c99c7c572d75ae4fa9ebc

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.63 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.0a202607230 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
28ENV 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:/bin0 B
29ENV 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/lib0 B
30ENV 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/lib640 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 JAX_VERSION=0.10.20 B
34ENV JAX_VERSION=0.10.20 B
35ENV JAX_PJRT_VERSION=0.10.2+rocm7.15.0a202607230 B
36ENV JAX_PLUGIN_VERSION=0.10.2+rocm7.15.0a202607230 B
37RUN |1 JAX_VERSION=0.10.2 /bin/sh -c286.99 MB
38ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a10 B
39RUN |2 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 /bin/sh364.73 MB
40ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B
41ENV NVTE_USE_HIPBLASLT=10 B
42ENV GPU_MAX_HW_QUEUES=20 B
43ENV HIP_FORCE_DEV_KERNARG=10 B
44ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B
45ENV NVTE_FUSED_ATTN=10 B
46ENV NCCL_DEBUG=VERSION0 B
47ENV NVTE_CK_USES_BWD_V3=10 B
48ENV NVTE_CK_USES_FWD_V3=10 B
49ENV NVTE_CK_IS_V3_ATOMIC_FP32=10 B
50ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
51ENV XLA_PYTHON_CLIENT_MEM_FRACTION=.90 B
52ENV 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=85899345920 B
53WORKDIR /workspace32 B
54ARG MAXTEXT_REPO=https://github.com/ROCm/maxtext.git0 B
55ARG MAXTEXT_BRANCH=release/v26.50 B
56RUN |4 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git1.48 GB
57ARG TF_REPO=https://github.com/ROCm/tensorflow-upstream.git0 B
58ARG TF_BRANCH=upstream-v2.21.00 B
59ARG BAZELISK_VERSION=v1.25.00 B
60RUN |7 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git3.01 MB
61RUN |7 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git264.61 MB
62ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
63ARG PRIMUS_BRANCH=main0 B
64RUN |9 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git1004.04 MB
65RUN |9 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git26.29 MB
66ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
67RUN |10 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git2.64 MB
68RUN |10 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git226 B
69ARG UCX_VERSION=1.18.00 B
70RUN |11 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git86.43 MB
71ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
72ARG MPI_VERSION=4.1.60 B
73RUN |12 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git3.84 MB
74RUN |12 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git443 B
75RUN |12 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git32 B
76ARG GIT_COMMIT_TAG=d28a27b5ee3caa872552d57e74ba95903043456f0 B
77ARG DOCKERFILE_PATH=jax_training_the_rock/docker/Dockerfile0 B
78RUN |14 JAX_VERSION=0.10.2 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a1 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git16.94 KB
79COPY jax_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit3.75 KB

Command

 ARG RELEASE