update base image to use PyTorch with CUDA support
Some checks failed
Build Worker Base and Application Images / build-docker (push) Blocked by required conditions
Build Worker Base and Application Images / deploy-stack (push) Blocked by required conditions
Build Worker Base and Application Images / check-base-changes (push) Successful in 8s
Build Worker Base and Application Images / build-base (push) Has been cancelled

This commit is contained in:
Siwat Sirichai 2025-08-31 00:55:02 +07:00
parent 715546cd90
commit 06270f04fc

View file

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