41 lines
946 B
Docker
41 lines
946 B
Docker
# DXA Quality Assessment - Docker Container
|
|
# Build: docker build -t dxa-quality .
|
|
# Run: docker run -v /path/to/data:/data -p 8000:8000 dxa-quality
|
|
|
|
# Base image with GPU support
|
|
FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 AS builder
|
|
|
|
# Install Python and build tools
|
|
RUN apt-get update && apt-get install -y \
|
|
python3.10 \
|
|
python3-pip \
|
|
build-essential \
|
|
meson \
|
|
ninja-build \
|
|
libgl1-mesa-glx \
|
|
libglib2.0-0 \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender1 \
|
|
libgomp1 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
WORKDIR /app
|
|
|
|
# Copy requirements and install Python dependencies
|
|
COPY requirements.txt .
|
|
RUN pip3 install --no-cache-dir -r requirements.txt
|
|
|
|
# Copy application code
|
|
COPY . .
|
|
RUN mkdir -p models
|
|
|
|
# Expose API port
|
|
EXPOSE 8000
|
|
|
|
# Environment variables
|
|
ENV PYTHONUNBUFFERED=1
|
|
ENV PYTHONPATH=/app
|
|
|
|
# Default command
|
|
CMD ["python3", "-m", "uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "8000"]
|