diff --git a/docker-compose-ollama-gpu.yaml b/docker-compose-ollama-gpu.yaml index 747e423..3093e90 100644 --- a/docker-compose-ollama-gpu.yaml +++ b/docker-compose-ollama-gpu.yaml @@ -33,7 +33,7 @@ services: reservations: devices: - driver: nvidia - count: 1 + count: all capabilities: [gpu] ollama-webui: