Grafikkarte im Container nutzen

Grafikkarte im Container nutzen

- Matt

Läuft Ollama im Container ohne Zusatz, rechnet es auf der CPU, auch wenn eine Grafikkarte vorhanden ist. Das erklärt die meisten Berichte über langsame lokale Modelle im Docker-Betrieb.

Nötig ist das NVIDIA Container Toolkit auf dem Host, danach:

services:
  ollama:
    image: ollama/ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

Prüfen lässt sich das von innen:

docker exec -it ollama nvidia-smi

Kommt dort eine Ausgabe, ist die Karte sichtbar. Ob das Modell sie auch benutzt, sagt danach ollama ps.