Sign inSign up

rajendra95/peft-trainer:axolotl-v1

Manifest digest

sha256:e40e29dd3f100bc1c30375d86cd516f9c78bc20b8f30f4c7e17b6347932205dd

OS/ARCH

linux/amd64

Compressed size

6.3 GB

Last pushed

1 day by rajendra95

Type

Image

Manifest digest

sha256:e40e29dd3f100bc1c30375d86cd516f9c78bc20b8f30f4c7e17b6347932205dd

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=24.040 B
5ADD file ... in / 28.35 MB
6CMD ["/bin/bash"]0 B
7ENV NVARCH=x86_640 B
8ENV NVIDIA_REQUIRE_CUDA=cuda>=13.0 brand=unknown,driver>=535,driver<536 brand=grid,driver>=535,driver<536 brand=tesla,driver>=535,driver<5360 B
9ENV NV_CUDA_CUDART_VERSION=13.0.88-10 B
10ARG TARGETARCH0 B
11LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B
12RUN |1 TARGETARCH=amd64 /bin/sh -c4.34 MB
13ENV CUDA_VERSION=13.0.10 B
14RUN |1 TARGETARCH=amd64 /bin/sh -c100.28 MB
15RUN |1 TARGETARCH=amd64 /bin/sh -c184 B
16ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
17ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib640 B
18COPY NGC-DL-CONTAINER-LICENSE / # buildkit6.72 KB
19ENV NVIDIA_VISIBLE_DEVICES=all0 B
20ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility0 B
21ENV NV_CUDA_LIB_VERSION=13.0.1-10 B
22ENV NV_NVTX_VERSION=13.0.85-10 B
23ENV NV_LIBNPP_VERSION=13.0.1.2-10 B
24ENV NV_LIBNPP_PACKAGE=libnpp-13-0=13.0.1.2-10 B
25ENV NV_LIBCUSPARSE_VERSION=12.6.3.3-10 B
26ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-13-00 B
27ENV NV_LIBCUBLAS_VERSION=13.0.2.14-10 B
28ENV NV_LIBCUBLAS_PACKAGE=libcublas-13-0=13.0.2.14-10 B
29ENV NV_LIBNCCL_PACKAGE_NAME=libnccl20 B
30ENV NV_LIBNCCL_PACKAGE_VERSION=2.28.3-10 B
31ENV NCCL_VERSION=2.28.3-10 B
32ENV NV_LIBNCCL_PACKAGE=libnccl2=2.28.3-1+cuda13.00 B
33ARG TARGETARCH0 B
34LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B
35RUN |1 TARGETARCH=amd64 /bin/sh -c1.4 GB
36RUN |1 TARGETARCH=amd64 /bin/sh -c58.22 KB
37COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit1.64 KB
38COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit1.49 KB
39ENV NVIDIA_PRODUCT_NAME=CUDA0 B
40ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B
41RUN --mount=type=cache,target=/var/cache/apt,sharing=locked 317.52 MB
42WORKDIR /app236 B
43RUN curl -LsSf https://astral.sh/uv/install.sh |25.39 MB
44RUN uv python install 3.1236.74 MB
45RUN uv venv --python 3.1212.68 MB
46RUN uv pip install torch==2.12.02.73 GB
47RUN uv pip install "flash-attn==2.8.3"1.07 GB
48RUN uv pip install "peft==0.20.0"592.41 MB
49RUN chmod -R a+rX "$VENV"5.43 KB
50COPY axolotl_entrypoint.py rackai_axolotl_plugin.py /app/9.59 KB

Command

 ARG RELEASE