From b730c683032bccba9f243f0d2274b0e32e670011 Mon Sep 17 00:00:00 2001
From: imp4ct <sean@corrgs.ca>
Date: Fri, 6 Dec 2024 13:09:54 -0500
Subject: [PATCH] nvidia gpu compat

---
 docker-compose.yml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index acdffae..579ad1e 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,7 +11,13 @@ services:
     image: ollama/ollama:latest
     ports:
       - 11434:11434
-
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
   ollama-webui:
     image: ghcr.io/ollama-webui/ollama-webui:main
     container_name: ollama-webui