diff --git a/docker-compose.yml b/docker-compose.yml
index dd61b0a..7c40ffb 100755
--- 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: