dotfiles/ollama/.config/ollama_container/compose_nvidia.yaml

15 lines
322 B
YAML
Raw Permalink Normal View History

2025-01-08 16:28:25 -05:00
services:
llama_server:
image: ollama/ollama:latest
container_name: llama_server
ports:
- "11434:11434"
volumes:
- ./container_data:/root/.ollama:z
privileged: true # Had to be done so nvidia can find gpus...
devices:
- nvidia.com/gpu=all
# - /dev/kfd
# - /dev/dri