From 0cbaf53e862ba78c92d79276515e236455423013 Mon Sep 17 00:00:00 2001
From: teddy <teddy.kim.kim@tf.dabeeo.com>
Date: Thu, 19 Mar 2026 17:08:37 +0900
Subject: [PATCH] =?UTF-8?q?=EC=8B=9C=EC=8A=A4=ED=85=9C=20=EC=82=AC?=
 =?UTF-8?q?=EC=9A=A9=EC=9C=A8=20=EB=AA=A8=EB=8B=88=ED=84=B0=EB=A7=81=20?=
 =?UTF-8?q?=EA=B8=B0=EB=8A=A5=20=EC=B6=94=EA=B0=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose-dev.yml  | 1 -
 docker-compose-prod.yml | 7 +++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker-compose-dev.yml b/docker-compose-dev.yml
index 1a38ed0..34b4d84 100644
--- a/docker-compose-dev.yml
+++ b/docker-compose-dev.yml
@@ -5,7 +5,6 @@ services:
       dockerfile: Dockerfile-dev
     image: kamco-cd-training-api:${IMAGE_TAG:-latest}
     container_name: kamco-cd-training-api
-    runtime: nvidia
     deploy:
       resources:
         reservations:
diff --git a/docker-compose-prod.yml b/docker-compose-prod.yml
index 4e796d8..a987358 100644
--- a/docker-compose-prod.yml
+++ b/docker-compose-prod.yml
@@ -5,6 +5,13 @@ services:
       dockerfile: Dockerfile
     image: kamco-train-api:${IMAGE_TAG:-latest}
     container_name: kamco-train-api
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     expose:
       - "8080"
     environment:
-- 
2.49.1