Update compose.yaml

This commit is contained in:
Timothy J. Baek 2023-10-30 13:12:33 -07:00
parent db4f673f99
commit fb94941541
1 changed files with 10 additions and 9 deletions

View File

@ -2,20 +2,21 @@ version: '3.6'
services:
ollama:
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities:
- gpu
# Comment below for GPU support
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# count: 1
# capabilities:
# - gpu
volumes:
- ollama:/root/.ollama
ports:
- 11434:11434
environment:
- OLLAMA_ORIGINS="*"
- 'OLLAMA_ORIGINS=*'
container_name: ollama
pull_policy: always
tty: true