瀏覽代碼

Add AMD docker compose file

Justin Hayes 1 年之前
父節點
當前提交
075dddf3d3
共有 1 個文件被更改,包括 8 次插入0 次删除
  1. 8 0
      docker-compose.amdgpu.yaml

+ 8 - 0
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}'