diff --git a/compose.yaml b/compose.yaml
index 71c2882..10adf2b 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -23,6 +23,7 @@ services:
       - /dev/nvidia0:/dev/nvidia0
       - /dev/nvidiactl:/dev/nvidiactl
       - /dev/nvidia-uvm:/dev/nvidia-uvm
+      - /dev/dri:/dev/dri
     ports:
       - "8096:8096"
     networks: