From 29bfb91727125ed67f825ff4cd8d5f327460b780 Mon Sep 17 00:00:00 2001 From: IronicBadger Date: Sun, 17 Mar 2024 18:48:20 -0700 Subject: [PATCH] gpu support --- templates/docker-compose.yml.j2 | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/templates/docker-compose.yml.j2 b/templates/docker-compose.yml.j2 index 31a06c5..97b2e79 100644 --- 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 }}