diff --git a/Dockerfile b/Dockerfile index 0cfcf4c..4bd2258 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,7 +8,7 @@ ARG VARIANT=gpu -FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04 AS base-amd64-gpu +FROM --platform=$BUILDPLATFORM nvidia/cuda:12.9.2-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