| ADD file:7fd52845d064446007966c062784c558669970932bd441501bc48bceaac68c35 in / |
| CMD ["/bin/bash"] |
| ENV NVARCH=x86_64 |
| ENV NVIDIA_REQUIRE_CUDA=cuda>=12.3 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526 brand=tesla,driver>=535,driver<536 brand=unknown,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=geforce,driver>=535,driver<536 brand=geforcertx,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=titan,driver>=535,driver<536 brand=titanrtx,driver>=535,driver<536 |
| ENV NV_CUDA_CUDART_VERSION=12.3.101-1 |
| COPY cuda.repo-x86_64 /etc/yum.repos.d/cuda.repo # buildkit |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <sw-cuda-installer@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c NVIDIA_GPGKEY_SUM=d0664fbbdb8c32356d45de36c5984617217b2d0bef41b93ccecd326ba3b80c87 && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel9/${NVARCH}/D42D0685.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && echo "$NVIDIA_GPGKEY_SUM /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c --strict - # buildkit |
| ENV CUDA_VERSION=12.3.1 |
| RUN |1 TARGETARCH=amd64 /bin/sh -c yum upgrade -y && yum install -y cuda-cudart-12-3-${NV_CUDA_CUDART_VERSION} cuda-compat-12-3 && yum clean all && rm -rf /var/cache/yum/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| COPY NGC-DL-CONTAINER-LICENSE / # buildkit |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV NV_CUDA_LIB_VERSION=12.3.1-1 |
| ENV NV_NVTX_VERSION=12.3.101-1 |
| ENV NV_LIBNPP_VERSION=12.2.3.2-1 |
| ENV NV_LIBNPP_PACKAGE=libnpp-12-3-12.2.3.2-1 |
| ENV NV_LIBCUBLAS_VERSION=12.3.4.1-1 |
| ENV NV_LIBNCCL_PACKAGE_NAME=libnccl |
| ENV NV_LIBNCCL_PACKAGE_VERSION=2.19.3-1 |
| ENV NV_LIBNCCL_VERSION=2.19.3 |
| ENV NCCL_VERSION=2.19.3 |
| ENV NV_LIBNCCL_PACKAGE=libnccl-2.19.3-1+cuda12.3 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <sw-cuda-installer@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c yum install -y cuda-libraries-12-3-${NV_CUDA_LIB_VERSION} cuda-nvtx-12-3-${NV_NVTX_VERSION} ${NV_LIBNPP_PACKAGE} libcublas-12-3-${NV_LIBCUBLAS_VERSION} ${NV_LIBNCCL_PACKAGE} && yum clean all && rm -rf /var/cache/yum/* # buildkit |
| COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit |
| COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit |
| ENV NVIDIA_PRODUCT_NAME=CUDA |
| ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"] |
| ENV NV_CUDA_LIB_VERSION=12.3.1-1 |
| ENV NV_NVPROF_VERSION=12.3.101-1 |
| ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-3-12.3.101-1 |
| ENV NV_CUDA_CUDART_DEV_VERSION=12.3.101-1 |
| ENV NV_NVML_DEV_VERSION=12.3.101-1 |
| ENV NV_LIBCUBLAS_DEV_VERSION=12.3.4.1-1 |
| ENV NV_LIBNPP_DEV_VERSION=12.2.3.2-1 |
| ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-3-12.2.3.2-1 |
| ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel |
| ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.19.3-1 |
| ENV NCCL_VERSION=2.19.3 |
| ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-devel-2.19.3-1+cuda12.3 |
| ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.3.1-1 |
| ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-3-12.3.1-1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <sw-cuda-installer@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c yum install -y make findutils cuda-command-line-tools-12-3-${NV_CUDA_LIB_VERSION} cuda-libraries-devel-12-3-${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-3-${NV_CUDA_LIB_VERSION} cuda-cudart-devel-12-3-${NV_CUDA_CUDART_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} cuda-nvml-devel-12-3-${NV_NVML_DEV_VERSION} libcublas-devel-12-3-${NV_LIBCUBLAS_DEV_VERSION} ${NV_LIBNPP_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} && yum clean all && rm -rf /var/cache/yum/* # buildkit |
| ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs |
| RUN /bin/sh -c dnf update -y && dnf install python3 python3-devel python3-pip cmake make gcc-c++ -y && dnf clean all -y # buildkit |
| WORKDIR /app |
| COPY ./pyproject.toml /app/pyproject.toml # buildkit |
| COPY ./poetry.lock /app/poetry.lock # buildkit |
| COPY ./sky /app/sky # buildkit |
| COPY ./worker /app/worker # buildkit |
| RUN /bin/sh -c pip install poetry # buildkit |
| RUN /bin/sh -c poetry install --only main # buildkit |
| RUN /bin/sh -c CMAKE_ARGS="-DLLAMA_CUBLAS=on" poetry run pip install llama-cpp-python --force-reinstall # buildkit |
| RUN /bin/sh -c poetry build # buildkit |
| CMD ["poetry" "run" "python" "sky/main.py"] |
