2026-03-19 17:09:08 +09:00
2 changed files with 7 additions and 1 deletions
--- 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:
--- 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: