Merge pull request #43 from edurenye/gpu-docker-compose

Issue #42 Add support for GPU in Docker compose Nice, thank you! :)
iot-salzburg · Feb 13, 2021 · f2873f9 · f2873f9
2 parents 488ebea + 5d88e30
commit f2873f9
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: