-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile
More file actions
42 lines (35 loc) · 1.35 KB
/
Dockerfile
File metadata and controls
42 lines (35 loc) · 1.35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# CPU (ONNX-CPU)
FROM python:3.12-slim-bullseye AS focoos_apps-cpu
COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
LABEL authors="focoos.ai"
RUN apt-get update --fix-missing && \
apt-get install -y --no-install-recommends \
build-essential git ffmpeg libsm6 libxext6 gcc libmagic1 wget make cmake python3.12-tk && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Copy focoos-apps and install
COPY pyproject.toml ./pyproject.toml
COPY focoos_apps ./focoos_apps
RUN uv pip install --system -e .[onnx-cpu]
# GPU (ONNX GPU)
FROM ghcr.io/focoosai/deeplearning:base-cu12-cudnn9-py312-uv AS focoos-apps-gpu
LABEL authors="focoos.ai"
RUN apt-get update --fix-missing && \
apt-get install -y --no-install-recommends python3.12-tk && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Copy focoos-apps and install
COPY pyproject.toml ./pyproject.toml
COPY focoos_apps ./focoos_apps
RUN uv pip install --system -e .[onnx-gpu]
# TensorRT (ONNX GPU + TensorRT)
FROM ghcr.io/focoosai/deeplearning:cu12-cudnn9-py312-uv-tensorrt AS focoos-apps-tensorrt
LABEL authors="focoos.ai"
RUN apt-get update --fix-missing && \
apt-get install -y --no-install-recommends python3.12-tk && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Copy focoos-apps and install
COPY pyproject.toml ./pyproject.toml
COPY focoos_apps ./focoos_apps
RUN uv pip install --system -e .[tensorrt]