coredotbin/ansible-role-docker-compose-generator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gpu support

Browse Source
This commit is contained in:
IronicBadger 2024-03-17 18:48:20 -07:00
parent 187bbd262f
commit 29bfb91727
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
templates/docker-compose.yml.j2
View File

@ -184,9 +184,9 @@ services:
resources: resources:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }} count: {{ container.deploy.resources.reservations.devices.driver.count | default("1") }}
capabilities: [gpu] capabilities: [gpu]
{% endif %} {% endif %}
{% if container.restart is defined %} {% if container.restart is defined %}
restart: {{ container.restart }} restart: {{ container.restart }}