diff --git a/docker-compose.yml b/docker-compose.yml index 4951fda..fd4dabc 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -143,5 +143,23 @@ services: reservations: devices: - device_ids: [ "1" ] + capabilities: [ "gpu" ] + driver: "nvidia" + + photo_review_6: + <<: *template + container_name: photo_review_6 + hostname: photo_review_6 + volumes: + - ./log:/app/log + - ./model:/app/model + depends_on: + - photo_mask_2 + command: [ "photo_review.py" ] + deploy: + resources: + reservations: + devices: + - device_ids: [ "0", "1" ] capabilities: [ "gpu" ] driver: "nvidia" \ No newline at end of file