/opt/nvidia/nvidia_entrypoint.sh

NVIDIA_PRODUCT_NAME=CUDA

nvidia_entrypoint.sh /opt/nvidia/

entrypoint.d/ /opt/nvidia/entrypoint.d/

TARGETARCH=arm64 apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME}

TARGETARCH=arm64 apt-get update &&
  apt-get install -y --no-install-recommends cuda-libraries-13-4=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-13-4=${NV_NVTX_VERSION} libcusparse-13-4=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} &&
  rm -rf /var/lib/apt/lists/*

maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

TARGETARCH

NV_LIBNCCL_PACKAGE=libnccl2=2.31.2-1+cuda13.4

NCCL_VERSION=2.31.2-1