Dockerfile

# Use Python 3.11 as base image
FROM python:3.11-slim

# Set working directory
WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    gcc \
    g++ \
    make \
    && rm -rf /var/lib/apt/lists/*

# Copy requirements first for better caching
COPY requirements.txt .

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt

# Copy the virtual GPU setup
COPY virtual_gpu_setup/ ./virtual_gpu_setup/

# Copy the application source code
COPY src/ ./src/

# Create necessary directories
RUN mkdir -p /app/src/static /app/src/templates

# Set environment variables
ENV PYTHONPATH=/app
ENV FLASK_APP=src.main
ENV FLASK_ENV=production

# Expose port 5000
EXPOSE 5000

# Health check
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
    CMD curl -f http://localhost:5000/api/gpu-status || exit 1

# Run the application
CMD ["python", "src/main.py"]