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
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:
parent
715546cd90
commit
06270f04fc
1 changed files with 1 additions and 1 deletions
|
@ -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/*
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue