# 使用官方的paddle镜像作为基础
FROM registry.baidubce.com/paddlepaddle/paddle:2.6.1-gpu-cuda12.0-cudnn8.9-trt8.6

# 设置工作目录
WORKDIR /app

# 设置环境变量
ENV PYTHONUNBUFFERED=1 \
    # 设置时区
    TZ=Asia/Shanghai \
    # 设置pip镜像地址，加快安装速度
    PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple

# 安装依赖
COPY requirements.txt /app/requirements.txt
COPY packages /app/packages
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
    && pip install --no-cache-dir -r requirements.txt \
    && pip uninstall -y onnxruntime onnxruntime-gpu \
    && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \
    && pip install packages/torch-2.4.0+cu118-cp310-cp310-linux_x86_64.whl \
    && pip install packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl \
    && pip install packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl

# 编译文档检测的部分内容
COPY object_detection /app/object_detection
RUN python object_detection/core/models/py_utils/_cpools/setup.py install --user  \
    && cd object_detection/core/external \
    && make

# 将当前目录内容复制到容器的/app内
COPY . /app

# 暴露端口
# EXPOSE 8081

# 运行python脚本，具体脚本在命令行或docker-compose.yml文件中定义
ENTRYPOINT ["python"]