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 apt-get update &&
  if apt-cache policy cuda-compat-orin-13-4 2>/dev/null | grep -q "Candidate:"; then apt-get install -y --no-install-recommends cuda-compat-orin-13-4; fi &&
  rm -rf /var/lib/apt/lists/*

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

TARGETARCH=arm64 IMAGE_UPGRADE_ID=20260914 apt-get update &&
  apt-get install -y --no-install-recommends 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} &&
  rm -rf /var/lib/apt/lists/*

CUDA_VERSION=13.4.1