new nvidia config

2022-04-29 16:19:25 +02:00 · 2022-04-29 16:19:25 +02:00 · 18974f5961
commit 18974f5961
parent 7b2475271f
1 changed files with 10 additions and 0 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@ -1,4 +1,14 @@
 services:
+  nvidia:
+    extends:
+      service: dev
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
  dev:
    network_mode: host
    build: