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
|
# 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/*
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue