diff --git a/templates/docker-compose.yml.j2 b/templates/docker-compose.yml.j2
index dd816af..31a06c5 100644
--- 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 %}