From 075dddf3d3e8492b8ff24c91d002065b73a1f015 Mon Sep 17 00:00:00 2001
From: Justin Hayes <justinh@rahb.ca>
Date: Fri, 5 Apr 2024 20:11:33 -0400
Subject: [PATCH] Add AMD docker compose file

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

diff --git a/docker-compose.amdgpu.yaml b/docker-compose.amdgpu.yaml
new file mode 100644
index 000000000..130b1ccb7
--- /dev/null
+++ b/docker-compose.amdgpu.yaml
@@ -0,0 +1,8 @@
+services:
+  ollama:
+    devices:
+      - /dev/kfd:/dev/kfd
+      - /dev/dri:/dev/dri
+    image: ollama/ollama:rocm
+    environment:
+      - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION}'
\ No newline at end of file