Entrypoint /opt/nvidia/nvidia_entrypoint.sh

TARGETARCH=amd64 apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME}

TARGETARCH=amd64 NSYS_PATH=$(find /opt/nvidia/nsight-compute -type f -name "nsys" 2>/dev/null | head -n 1); if [ -n "$NSYS_PATH" ] &&
  [ ! -f /usr/local/bin/nsys ]; then ln -s "$NSYS_PATH" /usr/local/bin/nsys; fi

TARGETARCH=amd64 apt-get update &&
  apt-get install -y --no-install-recommends cuda-cudart-dev-12-9=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-9=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-9=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-9=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-9=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-9=${NV_LIBCUSPARSE_DEV_VERSION} ${NV_LIBCUBLAS_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} &&
  rm -rf /var/lib/apt/lists/*

maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

TARGETARCH

NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.27.3-1+cuda12.9

NCCL_VERSION=2.27.3-1

NV_LIBNCCL_DEV_PACKAGE_VERSION=2.27.3-1

NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev