gpu support

2024-03-17 18:48:20 -07:00 · 2024-03-17 18:48:20 -07:00 · 29bfb91727
commit 29bfb91727
parent 187bbd262f
1 changed files with 3 additions and 3 deletions
--- a/templates/docker-compose.yml.j2
+++ b/templates/docker-compose.yml.j2
@ -184,9 +184,9 @@ services:
      resources:
        reservations:
          devices:
-            - driver: nvidia
-              count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }}
-              capabilities: [gpu]
+          - driver: nvidia
+            count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }}
+            capabilities: [gpu]
 {% endif %}
 {% if container.restart is defined %}
    restart: {{ container.restart }}