feat: allow jupyter to access to GPUs

sileht · sileht · commit 3448426e1946 · 2020-11-23T12:47:57.000+01:00
diff --git 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
