NVIDIA_DRIVER_CAPABILITIES=compute,utility

NVIDIA_VISIBLE_DEVICES=all

NGC-DL-CONTAINER-LICENSE /

LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64

PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260914 echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/nvidia.conf

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260914 echo "IMAGE_UPGRADE_ID=${IMAGE_UPGRADE_ID}" &&
  ${PKG_CMD} upgrade -y &&
  ${PKG_CMD} install -y cuda-cudart-13-4-${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-4-config-common-${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-config-common-${NV_CUDA_CUDART_VERSION} cuda-toolkit-config-common-${NV_CUDA_CUDART_VERSION} &&
  if ${PKG_CMD} info cuda-compat-13-4 2>/dev/null | grep -q "^Name"; then ${PKG_CMD} install -y cuda-compat-13-4; fi &&
  ${PKG_CMD} clean all &&
  rm -rf /var/cache/yum/*

IMAGE_UPGRADE_ID=0

TARGETARCH=arm64 UBI_ARCH=$([ "${NVARCH}" = "sbsa" ] &&
  echo "aarch64" ||
  echo "${NVARCH}") &&
  echo -e "[ubi-8-appstream]\nname=Red Hat Universal Base Image 8 (AppStream)\nbaseurl=https://cdn-ubi.redhat.com/content/public/ubi/dist/ubi8/8/${UBI_ARCH}/appstream/os\nenabled=1\ngpgcheck=1\ngpgkey=https://www.redhat.com/security/data/fd431d51.txt" > /etc/yum.repos.d/ubi-8-appstream.repo

CUDA_VERSION=13.4.1