Explorar el Código

Add AMD docker compose file

Justin Hayes hace 1 año
padre
commit
075dddf3d3
Se han modificado 1 ficheros con 8 adiciones y 0 borrados
  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}'