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 ${PKG_CMD} update --releasever latest -y

CUDA_VERSION=13.4.1