From 237dd2091b68f59b8196bc2d48a886a628cd48a0 Mon Sep 17 00:00:00 2001 From: Holden Date: Sun, 14 Jun 2026 19:17:13 +0000 Subject: [PATCH] build: bump amd64 gpu base to NVIDIA CUDA ubuntu24.04 Highest Ubuntu version NVIDIA currently publishes for CUDA 12.8.1. 26.04 not yet available from NVIDIA's image registry. --- Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index b25d5c9..08026cb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,14 +1,14 @@ # ── Base images ─────────────────────────────────────────────────────────────── -# amd64 + gpu: NVIDIA CUDA 12.8 + cuDNN (GPU acceleration via NVIDIA Container Toolkit) +# amd64 + gpu: NVIDIA CUDA 12.8 + cuDNN on Ubuntu 24.04 (highest Ubuntu NVIDIA publishes) # amd64 + rocm: Ubuntu 26.04 (AMD GPU via ROCm — pass /dev/kfd and /dev/dri) # amd64 + intel: Ubuntu 22.04 (Intel Arc / iGPU via OpenVINO — pass /dev/dri) -# Note: intel stays on 22.04 until Intel's GPU repo ships 26.04 packages +# Note: intel stays on 22.04 — Intel's GPU repo only publishes for jammy # amd64 + cpu: Ubuntu 26.04 (CPU-only, ~2 GB smaller image) # arm64: Ubuntu 24.04 (CPU-only; no CUDA/ROCm wheels on ARM) ARG VARIANT=gpu -FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu22.04 AS base-amd64-gpu +FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04 AS base-amd64-gpu FROM ubuntu:26.04 AS base-amd64-rocm FROM ubuntu:22.04 AS base-amd64-intel FROM ubuntu:26.04 AS base-amd64-cpu