From f33714861239ff2d84e584a0f81acb68eaf4664f Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Thu, 29 Aug 2024 08:50:22 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95docker=E6=9C=8D=E5=8A=A1?= =?UTF-8?q?=E4=B9=8B=E9=97=B4=E7=9A=84=E8=BF=9E=E9=80=9A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker-compose-test.yml | 42 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 docker-compose-test.yml diff --git a/docker-compose-test.yml b/docker-compose-test.yml new file mode 100644 index 0000000..9e28ecb --- /dev/null +++ b/docker-compose-test.yml @@ -0,0 +1,42 @@ +x-env: + &template + image: fcb_photo_review:1.13.8 + +services: + det_api: + <<: *template + build: + context: . + container_name: det_api + hostname: det_api + volumes: + - ./log:/app/log + - ./model:/app/model + command: [ "det_api.py" ] + ports: + - "5000:5000" # 映射端口 + deploy: + resources: + reservations: + devices: + - device_ids: [ "0" ] + capabilities: [ "gpu" ] + driver: "nvidia" + + photo_review_test: + <<: *template + container_name: photo_review_test + hostname: photo_review_test + volumes: + - ./log:/app/log + - ./model:/app/model + depends_on: + - det_api + command: [ "photo_review.py" ] + deploy: + resources: + reservations: + devices: + - device_ids: [ "0", "1" ] + capabilities: [ "gpu" ] + driver: "nvidia" \ No newline at end of file