Sign inSign up

rackspacengpc/rackai-trainer:1.1.0-dev.38e3c6-axolotl

Index digest

sha256:d0bdd86837d407c3e65fd81a2ac55462d5a4c91ca250faa6b074557701cb7144

OS/ARCH

linux/amd64

Compressed size

5.82 GB

Last pushed

16 days by rackspacengpc

Type

Image

Manifest digest

sha256:1babb61d4567d434c7aaacf083320acac0208ddb3056dc6d662293c1e31afc63

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
41ENV DEBIAN_FRONTEND=noninteractive0 B
42ENV PYTHONUNBUFFERED=10 B
43ENV USER=raxtrainer0 B
44RUN /bin/sh -c apt-get update249.04 MB
45WORKDIR /app93 B
46ENV UV_INSTALL_DIR=/usr/local/bin0 B
47RUN /bin/sh -c curl -LsSf19.3 MB
48ENV UV_PYTHON_INSTALL_DIR=/opt/uv-python0 B
49RUN /bin/sh -c uv python32.62 MB
50ENV VENV=/opt/venv0 B
51RUN /bin/sh -c uv venv9.49 KB
52ENV PATH=/opt/venv/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
53RUN /bin/sh -c uv pip2.59 GB
54ARG MAX_JOBS=40 B
55ARG NVCC_THREADS=60 B
56ENV MAX_JOBS=40 B
57ENV NVCC_THREADS=60 B
58ENV TORCH_CUDA_ARCH_LIST=9.00 B
59RUN |2 MAX_JOBS=4 NVCC_THREADS=6 /bin/sh931.76 MB
60ARG AXOLOTL_REF=93dcd622f1ceddaabce272569736ec0f205fba4d0 B
61RUN |3 MAX_JOBS=4 NVCC_THREADS=6 AXOLOTL_REF=93dcd622f1ceddaabce272569736ec0f205fba4d503.31 MB
62RUN |3 MAX_JOBS=4 NVCC_THREADS=6 AXOLOTL_REF=93dcd622f1ceddaabce272569736ec0f205fba4d93 B
63ENV HF_HOME=/app/.cache/huggingface0 B
64ENV TRANSFORMERS_CACHE=/app/.cache/huggingface0 B
65ENV TRITON_CACHE_DIR=/tmp/.triton0 B
66COPY axolotl_entrypoint.py rackai_axolotl_plugin.py /app/ #9.17 KB
67ENV PYTHONPATH=/app0 B
68CMD ["python3" "/app/axolotl_entrypoint.py"]0 B

Command

 ARG RELEASE