tensorfuse/axolotl-train:v0.0.10-test-v2

Manifest digest

sha256:fa0fe492aa77eee4b9138d29dbc03a180471b57eee475794d4cba569dc8a912c

OS/ARCH

linux/amd64

Compressed size

14.58 GB

Last pushed

over 1 year by tensorfuse

Type

Image

Manifest digest

sha256:fa0fe492aa77eee4b9138d29dbc03a180471b57eee475794d4cba569dc8a912c

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=22.040 B

5ADD file ... in / 28.17 MB

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

7ENV NVARCH=x86_640 B

8

ENV NVIDIA_REQUIRE_CUDA=cuda>=12.4 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471

0 B

9ENV NV_CUDA_CUDART_VERSION=12.4.127-10 B

10ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-40 B

11ARG TARGETARCH0 B

12LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B

13RUN |1 TARGETARCH=amd64 /bin/sh -c4.41 MB

14ENV CUDA_VERSION=12.4.10 B

15RUN |1 TARGETARCH=amd64 /bin/sh -c54.93 MB

16RUN |1 TARGETARCH=amd64 /bin/sh -c185 B

17ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

18ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B

19COPY NGC-DL-CONTAINER-LICENSE / # buildkit6.72 KB

20ENV NVIDIA_VISIBLE_DEVICES=all0 B

21ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility0 B

22ENV NV_CUDA_LIB_VERSION=12.4.1-10 B

23ENV NV_NVTX_VERSION=12.4.127-10 B

24ENV NV_LIBNPP_VERSION=12.2.5.30-10 B

25ENV NV_LIBNPP_PACKAGE=libnpp-12-4=12.2.5.30-10 B

26ENV NV_LIBCUSPARSE_VERSION=12.3.1.170-10 B

27ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-40 B

28ENV NV_LIBCUBLAS_VERSION=12.4.5.8-10 B

29ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-4=12.4.5.8-10 B

30ENV NV_LIBNCCL_PACKAGE_NAME=libnccl20 B

31ENV NV_LIBNCCL_PACKAGE_VERSION=2.21.5-10 B

32ENV NCCL_VERSION=2.21.5-10 B

33ENV NV_LIBNCCL_PACKAGE=libnccl2=2.21.5-1+cuda12.40 B

34ARG TARGETARCH0 B

35LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B

36RUN |1 TARGETARCH=amd64 /bin/sh -c1.28 GB

37RUN |1 TARGETARCH=amd64 /bin/sh -c62.55 KB

38COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit1.64 KB

39COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit1.49 KB

40ENV NVIDIA_PRODUCT_NAME=CUDA0 B

41ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]0 B

42ENV NV_CUDA_LIB_VERSION=12.4.1-10 B

43ENV NV_CUDA_CUDART_DEV_VERSION=12.4.127-10 B

44ENV NV_NVML_DEV_VERSION=12.4.127-10 B

45ENV NV_LIBCUSPARSE_DEV_VERSION=12.3.1.170-10 B

46ENV NV_LIBNPP_DEV_VERSION=12.2.5.30-10 B

47ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-4=12.2.5.30-10 B

48ENV NV_LIBCUBLAS_DEV_VERSION=12.4.5.8-10 B

49ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-40 B

50ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-4=12.4.5.8-10 B

51ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.4.1-10 B

52ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-4=12.4.1-10 B

53ENV NV_NVPROF_VERSION=12.4.127-10 B

54ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-4=12.4.127-10 B

55ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev0 B

56ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.21.5-10 B

57ENV NCCL_VERSION=2.21.5-10 B

58ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.21.5-1+cuda12.40 B

59ARG TARGETARCH0 B

60LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B

61RUN |1 TARGETARCH=amd64 /bin/sh -c2.46 GB

62RUN |1 TARGETARCH=amd64 /bin/sh -c86.78 KB

63ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs0 B

64ENV NV_CUDNN_VERSION=9.1.0.70-10 B

65ENV NV_CUDNN_PACKAGE_NAME=libcudnn9-cuda-120 B

66ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12=9.1.0.70-10 B

67ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-dev-cuda-12=9.1.0.70-10 B

68ARG TARGETARCH0 B

69LABEL maintainer=NVIDIA CORPORATION <[email protected]>0 B

70LABEL com.nvidia.cudnn.version=9.1.0.70-10 B

71RUN |1 TARGETARCH=amd64 /bin/sh -c639.13 MB

72

ENV PATH=/root/miniconda3/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

73ARG PYTHON_VERSION=3.110 B

74ARG PYTORCH_VERSION=2.6.00 B

75ARG CUDA=1240 B

76ARG TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0 8.60 B

77ENV PYTHON_VERSION=3.110 B

78ENV TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0 8.60 B

79RUN |4 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.6.0 CUDA=124502.02 MB

80

ENV PATH=/root/miniconda3/envs/py3.11/bin:/root/miniconda3/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

81WORKDIR /workspace99 B

82RUN |4 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.6.0 CUDA=1246.11 GB

83RUN |4 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.6.0 CUDA=12439.53 MB

84RUN |4 PYTHON_VERSION=3.11 PYTORCH_VERSION=2.6.0 CUDA=12432 B

85ARG TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0 8.6+PTX0 B

86ARG AXOLOTL_EXTRAS=vllm0 B

87ARG AXOLOTL_ARGS=0 B

88ARG CUDA=1240 B

89ARG PYTORCH_VERSION=2.6.00 B

90ENV PYTORCH_VERSION=2.6.00 B

91RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0662.34 MB

92WORKDIR /workspace32 B

93RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.04.89 MB

94WORKDIR /workspace/axolotl32 B

95RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.02.69 GB

96RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.032 B

97RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.04.2 MB

98RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.01.39 MB

99RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0353 B

100RUN |5 TORCH_CUDA_ARCH_LIST=7.0 7.5 8.0233 B

101RUN /bin/sh -c apt-get update17.61 MB

102RUN /bin/sh -c curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip"60.48 MB

103RUN /bin/sh -c pip3 install4.23 MB

104ENV HF_HUB_ENABLE_HF_TRANSFER=10 B

105WORKDIR /app92 B

106COPY main.py /app # buildkit2.48 KB

107COPY lora.py /app # buildkit736 B

108COPY requirements.txt /app # buildkit202 B

109COPY entrypoint.sh /app/entrypoint.sh # buildkit678 B

110RUN /bin/sh -c chmod +x673 B

111RUN /bin/sh -c pip3 install51 MB

112ENTRYPOINT ["./entrypoint.sh"]0 B

Command

 ARG RELEASE