[Docker] Update the CUDA version in the default Docker image to 12.8 (from 12.1) #3163

Author: peterschmidt85

docker/base/Dockerfile

@@ -2,7 +2,7 @@
 ARG UBUNTU_VERSION
 
 # Build stage
-FROM nvidia/cuda:12.1.1-base-ubuntu${UBUNTU_VERSION}.04 AS builder
+FROM nvidia/cuda:12.8.1-base-ubuntu${UBUNTU_VERSION}.04 AS builder
 
 ENV NCCL_HOME=/opt/nccl
 ENV CUDA_HOME=/usr/local/cuda

docker/base/Dockerfile.common

@@ -1,6 +1,6 @@
 ARG UBUNTU_VERSION
 
-FROM nvidia/cuda:12.1.1-base-ubuntu${UBUNTU_VERSION}.04
+FROM nvidia/cuda:12.8.1-base-ubuntu${UBUNTU_VERSION}.04
 
 ARG _UV_HOME="/opt/uv"
 

src/dstack/_internal/core/backends/vastai/compute.py

@@ -47,7 +47,7 @@ def __init__(self, config: VastAIConfig):
                     "reliability2": {"gte": 0.9},
                     "inet_down": {"gt": 128},
                     "verified": {"eq": True},
-                    "cuda_max_good": {"gte": 12.1},
+                    "cuda_max_good": {"gte": 12.8},
                     "compute_cap": {"gte": 600},
                 }
             )