Sign inSign up

rocm/jax-training:base-v26.6

Manifest digest

sha256:0995b15b02ece888dc61c03dd8355837da22e9b2eb65c79cf06f9a87ddc2d225

OS/ARCH

linux/amd64

Compressed size

7.35 GB

Last pushed

17 days by rocm

Type

Image

Manifest digest

sha256:0995b15b02ece888dc61c03dd8355837da22e9b2eb65c79cf06f9a87ddc2d225

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 update240.75 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
16COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
17RUN /bin/sh -c pip install32.84 MB
18ENV THE_ROCK_VERSION=7.14.00 B
19RUN /bin/sh -c python -m3.16 GB
20RUN /bin/sh -c rocm-sdk init3.16 GB
21COPY /bazelisk /usr/local/bin/bazel # buildkit5.67 MB
22ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
23ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
24ENV HIP_PLATFORM=amd0 B
25ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
27ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
28ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
29ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
30ENV 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
31ENV 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
32ENV 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
33ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
34ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
35RUN /bin/sh -c git clone45.54 MB
36ARG JAX_VERSION=0.11.00 B
37ENV JAX_VERSION=0.11.00 B
38ARG JAX_ROCM_VERSION=0.11.0.post10 B
39RUN |2 JAX_VERSION=0.11.0 JAX_ROCM_VERSION=0.11.0.post1 /bin/sh285.16 MB
40ARG TE_VERSION=2.17.0+rocm7.14.0.50a84ad0 B
41RUN |3 JAX_VERSION=0.11.0 JAX_ROCM_VERSION=0.11.0.post1 TE_VERSION=2.17.0+rocm7.14.0.50a84ad392.55 MB
42COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.24 MB

Command

 ARG RELEASE