LIBRARY_PATH=/usr/local/cuda/lib64/stubs

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

TARGETARCH=arm64 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=arm64 apt-get update &&
  apt-get install -y --no-install-recommends cuda-cudart-dev-13-4=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-13-4=${NV_CUDA_LIB_VERSION} cuda-minimal-build-13-4=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-13-4=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-13-4=${NV_NVML_DEV_VERSION} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-13-4=${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.31.2-1+cuda13.4

NCCL_VERSION=2.31.2-1

NV_LIBNCCL_DEV_PACKAGE_VERSION=2.31.2-1

NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev