FROM nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04

# 使用国内镜像源加速 apt
RUN sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list && \
    sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list

# 避免交互提示
ENV DEBIAN_FRONTEND=noninteractive

# 安装 Python 和系统依赖
RUN apt-get update && apt-get install -y \
    python3.12 \
    python3.12-dev \
    python3-pip \
    build-essential \
    libpq-dev \
    ffmpeg \
    libsm6 \
    libxext6 \
    && rm -rf /var/lib/apt/lists/*

# 设置 pip 使用国内镜像源（全局生效）
ENV PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
ENV PIP_TRUSTED_HOST=pypi.tuna.tsinghua.edu.cn

# 设置工作目录
WORKDIR /app

# 复制依赖文件并安装
COPY requirements.txt .
# 安装基础依赖，并添加GPU版本的onnxruntime 1.21.0
RUN python3.12 -m pip install --no-cache-dir -r requirements.txt onnxruntime-gpu==1.21.0

# 安装 PyTorch GPU 版本 (兼容CUDA 12.0)
RUN python3.12 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

# 复制应用代码
COPY . .

# 创建必要的目录
RUN mkdir -p app/static/uploads app/static/datasets app/static/models app/static/results app/static/annotations logs/tensorboard

# 暴露端口
EXPOSE 8000 6006

# 启动命令
CMD ["python3.12", "run.py"]