Sign inSign up

genesu100/utc_ci:33437732501

Index digest

sha256:7a1e94f4019e46722e968cca6d7234a01b4e052ff9f2403911d0492a0d9bbd59

OS/ARCH

linux/amd64

Compressed size

11.65 GB

Last pushed

about 1 month by genesu100

Type

Image

Manifest digest

sha256:b89b3f032b868d6c3e5887b4c06997696481bb8e48242a4a32251b5304825669

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.37 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 update245.11 MB
14COPY /uv /usr/local/bin/uv # buildkit20.96 MB
15RUN /bin/sh -c uv venv9.64 KB
16ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
17ENV VIRTUAL_ENV=/opt/venv0 B
18COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
19RUN /bin/sh -c uv pip29.14 MB
20ENV THE_ROCK_VERSION=10.1.0a202608220 B
21RUN /bin/sh -c uv pip3.63 GB
22RUN /bin/sh -c rocm-sdk init3.63 GB
23COPY /bazelisk /usr/local/bin/bazel # buildkit5.67 MB
24ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
25ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV HIP_PLATFORM=amd0 B
27ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
28ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
29ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
30ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
31ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
32ENV 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
33ENV 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
34ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
35ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
36ENV LD_LIBRARY_PATH=0 B
37ARG JAX_VERSION=0.11.00 B
38ENV JAX_VERSION=0.11.00 B
39ENV JAX_PJRT_VERSION=0.11.0+rocm10.1.0a202608220 B
40ENV JAX_PLUGIN_VERSION=0.11.0+rocm10.1.0a202608220 B
41RUN |1 JAX_VERSION=0.11.0 /bin/sh -c266.61 MB
42RUN |1 JAX_VERSION=0.11.0 /bin/sh -c2.1 KB
43ARG TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb00 B
44RUN |2 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 /bin/sh394.23 MB
45COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.24 MB
46ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B
47ENV NVTE_USE_HIPBLASLT=10 B
48ENV GPU_MAX_HW_QUEUES=20 B
49ENV HIP_FORCE_DEV_KERNARG=10 B
50ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B
51ENV NVTE_FUSED_ATTN=10 B
52ENV NCCL_DEBUG=VERSION0 B
53ENV NVTE_CK_USES_BWD_V3=10 B
54ENV NVTE_CK_USES_FWD_V3=10 B
55ENV NVTE_CK_IS_V3_ATOMIC_FP32=10 B
56ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
57ENV XLA_PYTHON_CLIENT_MEM_FRACTION=.90 B
58ENV 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
59WORKDIR /workspace32 B
60ARG MAXTEXT_REPO=https://github.com/ROCm/maxtext.git0 B
61ARG MAXTEXT_BRANCH=release/v26.60 B
62RUN |4 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git1.5 GB
63ARG TF_REPO=https://github.com/ROCm/tensorflow-upstream.git0 B
64ARG TF_BRANCH=upstream-v2.21.00 B
65RUN |6 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git507.39 KB
66RUN |6 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git254.41 MB
67ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
68ARG PRIMUS_BRANCH=main0 B
69RUN |8 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git1.4 GB
70RUN |8 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git110.57 MB
71COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.24 MB
72COPY /gcsfuse /usr/bin/gcsfuse # buildkit24.85 MB
73RUN |8 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git26.72 MB
74ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
75RUN |9 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git364.3 KB
76RUN |9 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git227 B
77ARG UCX_VERSION=1.18.00 B
78RUN |10 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git86.43 MB
79ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
80ARG MPI_VERSION=4.1.60 B
81RUN |11 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git3.84 MB
82RUN |11 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git541 B
83RUN |11 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git32 B
84ARG GIT_COMMIT_TAG=d7bd009cd3c9cd2936daf336aa86f0de3063c0360 B
85ARG DOCKERFILE_PATH=jax_training_the_rock/docker/Dockerfile0 B
86RUN |13 JAX_VERSION=0.11.0 TE_VERSION=2.17.0+rocm10.1.0a20260817.ce80fb0 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git15.97 KB
87COPY jax_training_the_rock/docker/Dockerfile /workspace/.manifest/docker-build-recipe.txt # buildkit10.06 KB

Command

 ARG RELEASE