From 18974f5961d88235748e46887e5a259591c2ff19 Mon Sep 17 00:00:00 2001
From: Thies Lennart Alff <lennart.alff@gmail.com>
Date: Fri, 29 Apr 2022 16:19:25 +0200
Subject: [PATCH] new nvidia config

---
 docker-compose.yml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index c1eefc3..121eeec 100644
--- 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: