diff --git a/docker-compose.yml b/docker-compose.yml index e89c7cb..3d097e8 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -148,21 +148,21 @@ services: capabilities: [ "gpu" ] driver: "nvidia" -# photo_review_6: -# image: fcb_photo_review -# container_name: photo_review_6 -# hostname: photo_review_6 -# volumes: -# - ./log:/app/log -# - ./model:/app/model -# restart: always -# depends_on: -# - photo_mask_3 -# command: [ "photo_review.py" ] -# deploy: -# resources: -# reservations: -# devices: -# - device_ids: [ "0", "1" ] -# capabilities: [ "gpu" ] -# driver: "nvidia" \ No newline at end of file + photo_review_6: + image: fcb_photo_review + container_name: photo_review_6 + hostname: photo_review_6 + volumes: + - ./log:/app/log + - ./model:/app/model + restart: always + depends_on: + - photo_mask_3 + command: [ "photo_review.py" ] + deploy: + resources: + reservations: + devices: + - device_ids: [ "0", "1" ] + capabilities: [ "gpu" ] + driver: "nvidia" \ No newline at end of file