chore: upgrade CUDA base image from 12.1.0 to 12.6.0

Author: federicopalma-pro

docker/dockerfile

@@ -1,5 +1,5 @@
 # Stage 1: Builder Stage - Install dependencies and build wheels if necessary
-FROM nvidia/cuda:12.1.0-base-ubuntu22.04 AS builder
+FROM nvidia/cuda:12.6.0-base-ubuntu22.04 AS builder
 
 # Prevent apt-get prompts
 ENV DEBIAN_FRONTEND=noninteractive
@@ -39,7 +39,7 @@ RUN mkdir -p /app/models /app/logs
 # ---
 
 # Stage 2: Final Stage - Setup runtime environment
-FROM nvidia/cuda:12.1.0-base-ubuntu22.04
+FROM nvidia/cuda:12.6.0-base-ubuntu22.04
 
 # Prevent apt-get prompts
 ENV DEBIAN_FRONTEND=noninteractive