2025-09-09 12:33:37 +00:00
1 changed files with 1 additions and 1 deletions
--- 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/*