Add AMD docker compose file

075dddf3 · Justin Hayes · GitHub · 7a7d1931 · 075dddf3
Unverified Commit 075dddf3 authored Apr 05, 2024 by Justin Hayes Committed by GitHub Apr 05, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

docker-compose.amdgpu.yaml docker-compose.amdgpu.yaml +8 -0

No files found.
--- a/docker-compose.amdgpu.yaml
+++ b/docker-compose.amdgpu.yaml
+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