Improved jupyter container

2023-03-05 10:14:13 -08:00
parent 80feb0690c
commit 41ee2dc5b7
2 changed files with 11 additions and 1 deletions
@@ -13,3 +13,13 @@ services:
    # host:container
    ports:
      - "127.0.0.1:8888:8888"
+
+    # Allow GPU access.
+    # Requires nvidia-containter-toolkit
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]