From 5be4c645030003825e59e886acab71a41366fb2e Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Mon, 30 Oct 2023 12:52:45 -0700
Subject: [PATCH] Create compose.yaml

---
 compose.yaml | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 compose.yaml

diff --git a/compose.yaml b/compose.yaml
new file mode 100644
index 00000000..94601958
--- /dev/null
+++ b/compose.yaml
@@ -0,0 +1,38 @@
+version: '3.6'
+
+services:
+  ollama:
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities:
+                - gpu
+    volumes:
+      - ollama:/root/.ollama
+    ports:
+      - 11434:11434
+    environment:
+      - OLLAMA_ORIGINS="*"
+    container_name: ollama
+    pull_policy: always
+    tty: true
+    restart: unless-stopped
+    image: ollama/ollama:latest
+
+  ollama-webui:
+    restart: unless-stopped
+    build:
+      context: .
+      args:
+        OLLAMA_API_BASE_URL: ''
+      dockerfile: Dockerfile
+    image: ollama-webui:latest
+    container_name: ollama-webui
+    ports:
+      - 3000:8080
+
+volumes:
+  ollama: {}