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

added gpu stuff

Browse Source
This commit is contained in:
IronicBadger 2024-03-17 18:45:25 -07:00
parent 379b9d4255
commit 187bbd262f
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -179,6 +179,15 @@ services:
{% endif %} {% endif %}
start_period: {{ container.healthcheck.start_period }} start_period: {{ container.healthcheck.start_period }}
{% endif %} {% 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 %} {% if container.restart is defined %}
restart: {{ container.restart }} restart: {{ container.restart }}
{% endif %} {% endif %}