Update compose.yaml

This commit is contained in:
Timothy J. Baek 2023-10-30 13:12:33 -07:00
parent db4f673f99
commit fb94941541

View file

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