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=20260715 echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/nvidia.conf

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260715 apt-get update &&
  if apt-cache policy cuda-compat-orin-13-3 2>/dev/null | grep -q "Candidate:"; then apt-get install -y --no-install-recommends cuda-compat-orin-13-3; fi &&
  rm -rf /var/lib/apt/lists/*

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260715 apt-get update &&
  if apt-cache policy cuda-compat-13-3 2>/dev/null | grep -q "Candidate:"; then apt-get install -y --no-install-recommends cuda-compat-13-3; fi &&
  rm -rf /var/lib/apt/lists/*

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260715 apt-get update &&
  apt-get install -y --no-install-recommends cuda-cudart-13-3=${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-3-config-common=${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-config-common=${NV_CUDA_CUDART_VERSION} cuda-toolkit-config-common=${NV_CUDA_CUDART_VERSION} &&
  rm -rf /var/lib/apt/lists/*

CUDA_VERSION=13.3.1