From 18d30a1235ff94c3a3ce69a6b3deab8b1995d1af Mon Sep 17 00:00:00 2001
From: AngryBeeSec <85034322+AngryBeeSec@users.noreply.github.com>
Date: Wed, 6 Oct 2021 20:50:50 -0400
Subject: [PATCH] Update docker-compose.yml

Should fix GPU issues on Arch based systems.
---
 docker-cuda/docker-compose.yml | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/docker-cuda/docker-compose.yml b/docker-cuda/docker-compose.yml
index 960f2684..91ec8177 100644
--- a/docker-cuda/docker-compose.yml
+++ b/docker-cuda/docker-compose.yml
@@ -1,4 +1,4 @@
-version: "2"
+version: "3.2"
 services:
   koboldai:
     build: .
@@ -10,7 +10,16 @@ services:
       - ../:/content/
       - $HOME/.Xauthority:/home/micromamba/.Xauthority:rw
     devices:
-      - /dev/kfd
       - /dev/dri
+      - /dev/nvidia0:/dev/nvidia0
+      - /dev/nvidiactl:/dev/nvidiactl
+      - /dev/nvidia-uvm:/dev/nvidia-uvm
+    deploy:
+      resources:
+        reservations:
+          devices:
+          - driver: nvidia
+            device_ids: ['0']
+            capabilities: [gpu]
     group_add:
       - video