added gpu stuff

2024-03-17 18:45:25 -07:00 · 2024-03-17 18:45:25 -07:00 · 187bbd262f
commit 187bbd262f
parent 379b9d4255
1 changed files with 9 additions and 0 deletions
--- a/templates/docker-compose.yml.j2
+++ b/templates/docker-compose.yml.j2
@ -179,6 +179,15 @@ services:
 {% endif %}
      start_period: {{ container.healthcheck.start_period }}
 {% endif %}
+{% if container.deploy.resources.reservations.devices is defined %}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }}
+              capabilities: [gpu]
+{% endif %}
 {% if container.restart is defined %}
    restart: {{ container.restart }}
 {% endif %}