Sign inSign up

kubeflow/torchtune-trainer:v2.3.0-rc.3

Multi-platform
Index digest

sha256:744a41b7dbfb7e42d89358c5fa027c7414b525c12bb062bb2f458af013877cd3

OS/ARCH

Compressed size

4.54 GB

Last pushed

2 months by kubeflow

Type

Image

Manifest digest

sha256:20ec0749acd3e36a981a3707945e311519b4638ff93643107a78a5d53f3b9c1d

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 29.04 MB
6CMD ["/bin/bash"]0 B
7ARG PYTORCH_VERSION=2.9.10 B
8ARG TRITON_VERSION=0 B
9ARG TARGETPLATFORM=linux/amd640 B
10ARG CUDA_VERSION=12.8.10 B
11LABEL com.nvidia.volumes.needed=nvidia_driver0 B
12RUN |4 PYTORCH_VERSION=2.9.1 TRITON_VERSION= TARGETPLATFORM=linux/amd646.87 MB
13COPY /opt/conda /opt/conda # buildkit4.11 GB
14RUN |4 PYTORCH_VERSION=2.9.1 TRITON_VERSION= TARGETPLATFORM=linux/amd6432 B
15ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
16ENV NVIDIA_VISIBLE_DEVICES=all0 B
17ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility0 B
18ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B
19ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
20ENV PYTORCH_VERSION=2.9.10 B
21WORKDIR /workspace99 B
22WORKDIR /workspace32 B
23RUN /bin/sh -c apt update86.15 MB
24COPY cmd/trainers/torchtune/requirements.txt . # buildkit209 B
25RUN /bin/sh -c pip install320.22 MB

Command

 ARG RELEASE