diff --git a/docker-compose.yaml b/docker-compose.yaml
index e57e732..f0f60cd 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -14,6 +14,12 @@ services:
       - ./jupyter/overrides.json:/opt/conda/share/jupyter/lab/settings/overrides.json
     environment:
       - JUPYTER_TOKEN=12345
+    devices:
+      - /dev/nvidia-uvm
+      - /dev/nvidia-uvm-tools
+      - /dev/nvidia-modeset
+      - /dev/nvidiactl
+      - /dev/nvidia0
     deploy:
       resources:
         reservations: