diff --git a/docker-compose.yml b/docker-compose.yml index 2946f81..decf457 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -152,21 +152,21 @@ services: capabilities: [ "gpu" ] driver: "nvidia" -# photo_review_6: -# <<: *default-image -# 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: + <<: *default-image + 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