diff --git a/code/gpu/docker-compose.yml b/code/gpu/docker-compose.yml
index 0894200..e5a9cd9 100644
--- a/code/gpu/docker-compose.yml
+++ b/code/gpu/docker-compose.yml
@@ -54,6 +54,7 @@ services:
 
   jupyter:
     image: jolibrain/jupyter_dd_notebook:${DD_JUPYTER_TAG}
+    runtime: nvidia
     user: root
     environment:
       - JUPYTER_LAB_ENABLE=yes