diff --git a/Dockerfile b/Dockerfile index e5023fc..fad6885 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,7 +2,7 @@ # amd64: NVIDIA CUDA 12.6 (GPU acceleration when available, CPU fallback) # arm64: Ubuntu 26.04 (CPU-only; no CUDA on ARM) -FROM --platform=$BUILDPLATFORM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu22.04 AS base-amd64 +FROM --platform=$BUILDPLATFORM nvidia/cuda:13.3.0-cudnn-runtime-ubuntu22.04 AS base-amd64 FROM --platform=$BUILDPLATFORM ubuntu:26.04 AS base-arm64 # ── Build stage ───────────────────────────────────────────────────────────