Sign inSign up

rocm/jax-training:base-v26.8

Manifest digest

sha256:0b5e55c917b316e6225d3c4aa9499a7ea6e1c7e407188cc096b19aded98e8878

OS/ARCH

linux/amd64

Compressed size

8.44 GB

Last pushed

5 days by rocm

Type

Image

Manifest digest

sha256:0b5e55c917b316e6225d3c4aa9499a7ea6e1c7e407188cc096b19aded98e8878

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.39 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
13ARG LIBC_SECURITY_REFRESH=2026-10-010 B
14RUN |1 LIBC_SECURITY_REFRESH=2026-10-01 /bin/sh -c239.22 MB
15COPY /uv /usr/local/bin/uv # buildkit20.96 MB
16RUN |1 LIBC_SECURITY_REFRESH=2026-10-01 /bin/sh -c9.64 KB
17ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
18ENV VIRTUAL_ENV=/opt/venv0 B
19COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
20RUN |1 LIBC_SECURITY_REFRESH=2026-10-01 /bin/sh -c29.26 MB
21ENV THE_ROCK_VERSION=10.2.0a202609230 B
22RUN |1 LIBC_SECURITY_REFRESH=2026-10-01 /bin/sh -c3.72 GB
23RUN |1 LIBC_SECURITY_REFRESH=2026-10-01 /bin/sh -c3.72 GB
24COPY /bazelisk /usr/local/bin/bazel # buildkit5.67 MB
25ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
27ENV HIP_PLATFORM=amd0 B
28ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
29ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
30ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
31ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
32ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
33ENV 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
34ENV 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
35ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
36ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
37ENV LD_LIBRARY_PATH=0 B
38ARG JAX_VERSION=0.11.10 B
39ENV JAX_VERSION=0.11.10 B
40ENV JAX_PJRT_VERSION=0.11.1+rocm10.2.0a202609230 B
41ENV JAX_PLUGIN_VERSION=0.11.1+rocm10.2.0a202609230 B
42RUN |2 LIBC_SECURITY_REFRESH=2026-10-01 JAX_VERSION=0.11.1 /bin/sh275.89 MB
43ARG TE_VERSION=2.18.0.dev0+rocm10.2.0a20260929.03afd8f0 B
44RUN |3 LIBC_SECURITY_REFRESH=2026-10-01 JAX_VERSION=0.11.1 TE_VERSION=2.18.0.dev0+rocm10.2.0a20260929.03afd8f400.37 MB
45COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.92 MB

Command

 ARG RELEASE