Issue iot-salzburg#42 Add support for GPU in Docker compose

edurenye · Feb 13, 2021 · 5d88e30 · 5d88e30
1 parent 488ebea
commit 5d88e30
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1,7 +1,13 @@
-version: "3.4"
+version: "3.8"
 services:
   gpu-jupyter:
     build: .build
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities:
+              - gpu
     ports:
       - ${JUPYTER_PORT}:8888
     volumes: