From 06270f04fce81b805784fb4f39f07f7570f6d464 Mon Sep 17 00:00:00 2001
From: Siwat Sirichai <siwat@siwatinc.com>
Date: Sun, 31 Aug 2025 00:55:02 +0700
Subject: [PATCH] update base image to use PyTorch with CUDA support

---
 Dockerfile.base | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.base b/Dockerfile.base
index 3700920..d1a1eec 100644
--- a/Dockerfile.base
+++ b/Dockerfile.base
@@ -1,5 +1,5 @@
 # Base image with all ML dependencies
-FROM python:3.13-bookworm
+FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
 
 # Install system dependencies
 RUN apt update && apt install -y libgl1 && rm -rf /var/lib/apt/lists/*