root 1 anno fa
parent
commit
8424b5663d
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      deploy/docker_compose/Dockerfile

+ 1 - 0
deploy/docker_compose/Dockerfile

@@ -1,5 +1,6 @@
 # FROM python:3.10.0a6-buster
 # FROM python:3.10.0a6-buster
 FROM paddlecloud/paddleocr:2.5-cpu-latest
 FROM paddlecloud/paddleocr:2.5-cpu-latest
+FROM paddlepaddle/paddle:2.5.0-gpu-cuda12.0-cudnn8.9-trt8.6
 ARG work_dir=/app
 ARG work_dir=/app
 ENV FLASK_APP_PATH=${work_dir}/server.py
 ENV FLASK_APP_PATH=${work_dir}/server.py
 ENV FLASK_PORT=5000
 ENV FLASK_PORT=5000