andreyvelichkevich/llama-trainer:latest

Index digest

sha256:3cfa8796e70355c9da747863053bbb714e7fc1dd047d25d6ec8625271d6f2724

OS/ARCH

linux/amd64

Compressed size

3.3 GB

Last pushed

Type

Image

Manifest digest

sha256:e69bf04ef539a4b8528c63e094195ee3898875db8a21f0c5b95bba74fece2f5a

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.03 MB

6CMD ["/bin/bash"]0 B

7ARG PYTORCH_VERSION=2.5.00 B

8ARG TRITON_VERSION=0 B

9ARG TARGETPLATFORM=linux/amd640 B

10ARG CUDA_VERSION=12.4.10 B

11LABEL com.nvidia.volumes.needed=nvidia_driver0 B

12RUN |4 PYTORCH_VERSION=2.5.0 TRITON_VERSION= TARGETPLATFORM=linux/amd646.88 MB

13COPY /opt/conda /opt/conda # buildkit3.07 GB

14RUN |4 PYTORCH_VERSION=2.5.0 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

19

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

20ENV PYTORCH_VERSION=2.5.00 B

21WORKDIR /workspace99 B

22RUN /bin/sh -c mkdir -p138 B

23RUN /bin/sh -c mkdir -p121 B

24WORKDIR /workspace32 B

25ADD requirements.txt /workspace/requirements.txt # buildkit238 B

26RUN /bin/sh -c pip install196.39 MB

27COPY src/ /workspace/src/ # buildkit9.31 KB

28RUN /bin/sh -c chgrp -R1.68 KB

29ENTRYPOINT ["torchrun" "/workspace/src/llama_recipes/finetuning.py"]0 B

Command

 ARG RELEASE