From d96897cb3476cbc9a9ad883a892deb37f67e27af Mon Sep 17 00:00:00 2001
From: Erik Stambaugh <erik@erikstambaugh.com>
Date: Tue, 20 Feb 2024 14:35:32 -0800
Subject: [PATCH] Add nvidia devices to compose file

---
 docker-compose.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

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: