build: 修改docker 构建流程

This commit is contained in:
2026-04-10 11:25:00 +08:00
parent 296db909aa
commit cb0e7f2a80
3 changed files with 45 additions and 24 deletions
+10 -18
View File
@@ -1,49 +1,41 @@
# AstrAI Dockerfile
# Multi-stage build for optimized image size
# AstrAI Dockerfile - Minimal
# Build stage
FROM python:3.12-slim AS builder
WORKDIR /app
# Install build dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
&& rm -rf /var/lib/apt/lists/*
# Copy project files first for version extraction
COPY astrai/ ./astrai/
COPY pyproject.toml .
# Install dependencies
RUN pip install --no-cache-dir --upgrade pip \
&& pip install --no-cache-dir .[dev]
&& pip install --no-cache-dir .
# Production stage
FROM python:3.12-slim AS production
FROM nvidia/cuda:12.6.0-runtime-ubuntu22.04 AS production
WORKDIR /app
# Install runtime dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 \
libpython3.12 \
&& rm -rf /var/lib/apt/lists/*
# Copy installed packages from builder
COPY --from=builder /usr/local/lib/python3.12/site-packages /usr/local/lib/python3.12/site-packages
COPY --from=builder /usr/local/bin /usr/local/bin
# Copy application code
COPY astrai/ ./astrai/
COPY scripts/tools/ ./scripts/tools/
COPY scripts/ ./scripts/
COPY assets/ ./assets/
COPY pyproject.toml .
COPY README.md .
# Create non-root user
RUN useradd -m -u 1000 astrai && chown -R astrai:astrai /app
USER astrai
# Set environment variables
ENV PYTHONUNBUFFERED=1
ENV PYTHONDONTWRITEBYTECODE=1
# Default command
CMD ["python", "-m", "astrai.inference.server"]
ENV PYTHONUNBUFFERED=1 \
PYTHONDONTWRITEBYTECODE=1