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

CUDA_VERSION=13.4.1

TARGETARCH=arm64 NVIDIA_GPGKEY_SUM=d0664fbbdb8c32356d45de36c5984617217b2d0bef41b93ccecd326ba3b80c87 &&
  curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/azl3/${NVARCH}/D42D0685.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA &&
  echo "$NVIDIA_GPGKEY_SUM  /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c --strict -