From 79372af947b12b6c1e36c64556987baff0a8fe07 Mon Sep 17 00:00:00 2001
From: Ronghang Hu <ronghang.hu@gmail.com>
Date: Thu, 3 Oct 2024 15:33:19 -0700
Subject: [PATCH] [demo] add GPU to resources

This small PR adds GPU specification in `docker-compose.yaml` for the SAM 2 interactive webdemo, following https://docs.docker.com/compose/how-tos/gpu-support/#example-of-a-compose-file-for-running-a-service-with-access-to-1-gpu-device.

(Close https://github.com/facebookresearch/sam2/issues/354)
---
 docker-compose.yaml | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 73b8eabf..7a5395a5 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -33,3 +33,10 @@ services:
       - VIDEO_ENCODE_MAX_WIDTH=1280
       - VIDEO_ENCODE_MAX_HEIGHT=720
       - VIDEO_ENCODE_VERBOSE=False
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]