gpu support

ironicbadger · Mar 18, 2024 · 29bfb91 · 29bfb91
1 parent 187bbd2
commit 29bfb91
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 }}