1
0
Fork 0
local-llm/docker-compose.nvidia.yml

25 lines
455 B
YAML
Raw Normal View History

include:
- docker-compose.base.yml
services:
# Begin Ollama service
ollama:
2024-03-28 18:00:29 +00:00
image: ollama/ollama:latest
2024-11-13 04:27:20 +00:00
ports:
- 11434:11434
# begin for NVIDIA GPU support
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
# end of section for NVIDIA GPU support
2024-11-13 04:27:20 +00:00
volumes:
2024-11-13 04:27:20 +00:00
- ollama_data:/root/.ollama
volumes:
ollama_data: