fix: update tests Docker image

leruaa · leruaa · commit 6be009887ce1 · 2025-06-12T09:43:38.000-07:00
diff --git a/Dockerfile.server b/Dockerfile.server
@@ -1,5 +1,5 @@
 # Start with a base image that includes CUDA
-FROM nvidia/cuda:12.5.1-devel-ubuntu22.04 AS builder
+FROM nvidia/cuda:12.9.0-devel-ubuntu22.04 AS builder
 
 # Install necessary packages
 RUN apt-get update && apt-get install -y \
diff --git a/Dockerfile.toolchain b/Dockerfile.toolchain
@@ -1,5 +1,5 @@
 # Start with a base image that includes CUDA
-FROM nvidia/cuda:12.5.1-devel-ubuntu22.04 AS builder
+FROM nvidia/cuda:12.0.0-devel-ubuntu22.04 AS builder
 
 # Install necessary packages
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC  apt-get install -y \
