adsist-cms/python-detector-worker
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

refactor: update base image to runtime version and install minimal CUDA development tools for FFmpeg
Some checks failed
Build Worker Base and Application Images / check-base-changes (push) Successful in 7s
Details
Build Worker Base and Application Images / build-base (push) Failing after 3m52s
Details
Build Worker Base and Application Images / build-docker (push) Has been skipped
Details
Build Worker Base and Application Images / deploy-stack (push) Has been skipped
Details

Browse source
This commit is contained in:
Siwat Sirichai 2025-09-26 00:48:39 +07:00
parent 6fe4b6ebf0
commit fa3ab5c6d2
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
Dockerfile.base
View file

@ -1,5 +1,5 @@
# Base image with complete ML and hardware acceleration stack
FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-devel
FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
# Install build dependencies and system libraries
RUN apt-get update && apt-get install -y \
@ -37,7 +37,13 @@ RUN apt-get update && apt-get install -y \
python3-numpy \
&& rm -rf /var/lib/apt/lists/*
# CUDA development tools already available in devel image
# Install minimal CUDA development tools (just what we need for FFmpeg)
RUN apt-get update && apt-get install -y \
cuda-nvcc-12-6 \
cuda-cudart-dev-12-6 \
libnvidia-encode-12-6 \
libnvidia-decode-12-6 \
&& rm -rf /var/lib/apt/lists/*
# Ensure CUDA paths are available
ENV PATH="/usr/local/cuda/bin:${PATH}"