Entrypoint /opt/nvidia/nvidia_entrypoint.sh

TARGETARCH=arm64 rm /etc/apt/sources.list.d/cuda.list

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

TARGETARCH=arm64 apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}

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

TARGETARCH=arm64 apt-get update &&
  apt upgrade -y &&
  apt-get install -y --no-install-recommends gnupg2 curl ca-certificates &&
  curl -fsSL https://kitmaker-web.nvidia.com/kitpicks/cuda-r12-2-tegra/12.2.12/005/repos/ubuntu2204/${NVARCH}/3bf863cc.pub | apt-key add - &&
  echo "deb https://kitmaker-web.nvidia.com/kitpicks/cuda-r12-2-tegra/12.2.12/005/repos/ubuntu2204/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list &&
  apt-get purge --autoremove -y curl

maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

TARGETARCH

NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-2=12.2.5.6-1

NV_LIBCUBLAS_DEV_VERSION=12.2.5.6-1