shinezhang66/esp_dl_env repository overview

# dockerfile
# 指定基础镜像为 Python 3.7
FROM python:3.7

# 设置工作目录
WORKDIR /app

# 创建 workspace 目录
RUN mkdir -p /app/workspace

# 复制应用程序代码到容器中
COPY app.py .  

# 安装依赖
RUN pip install numba==0.53.1 onnx==1.9.0 onnxruntime==1.7.0 onnxoptimizer==0.2.6 flask gunicorn

# 设置环境变量
ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
# 设置 Flask 应用的入口点
ENV FLASK_APP=app.py  

# 验证 Numba 版本
RUN python3 -c "import numba; print('Numba version:', numba.__version__) if hasattr(numba, '__version__') else print('Error importing Numba: No version found')"

# 验证 ONNX 版本
RUN python3 -c "import onnx; print('ONNX version:', onnx.__version__) if hasattr(onnx, '__version__') else print('Error importing ONNX: No version found')"

# 验证 ONNX Runtime 版本
RUN python3 -c "import onnxruntime; print('ONNX Runtime version:', onnxruntime.__version__) if hasattr(onnxruntime, '__version__') else print('Error importing ONNX Runtime: No version found')"

# 验证 ONNX Optimizer 版本
RUN python3 -c "import onnxoptimizer; print('ONNX Optimizer version:', onnxoptimizer.__version__) if hasattr(onnxoptimizer, '__version__') else print('Error importing ONNX Optimizer: No version found')"

# 设置默认命令，使用 Gunicorn 运行 Flask 应用
CMD ["gunicorn", "-b", "0.0.0.0:8000", "app:app"]