adds cdi driver support for GPUs

ironicbadger · Sep 9, 2024 · a80f0ae · a80f0ae
1 parent 447db55
commit a80f0ae
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/templates/docker-compose.yml.j2 b/templates/docker-compose.yml.j2
@@ -185,9 +185,22 @@ services:
       resources:
         reservations:
           devices:
-          - driver: nvidia
-            count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }}
-            capabilities: [gpu]
+{% for device in container.deploy.resources.reservations.devices %}
+            - driver: {{ device.driver }}
+{% if device.device_ids is defined %}
+              device_ids:
+{% for device_id in device.device_ids %}
+                - {{ device_id }}
+{% endfor %}
+{% endif %}
+{% endfor %}
+{% else %}
+    deploy:
+        resources:
+        reservations:
+            devices:
+            - count: {{ container.deploy.resources.reservations.devices[0].driver.count | default("1") }}
+                capabilities: [gpu]
 {% endif %}
 {% if container.restart is defined %}
     restart: {{ container.restart }}