Switching to NVIDIA processing...
This commit is contained in:
@@ -11,8 +11,18 @@ services:
|
|||||||
- ollama:/root/.ollama
|
- ollama:/root/.ollama
|
||||||
|
|
||||||
environment:
|
environment:
|
||||||
# Keep model in memory a bit (optional)
|
|
||||||
OLLAMA_KEEP_ALIVE: "10m"
|
OLLAMA_KEEP_ALIVE: "10m"
|
||||||
|
# Optional: pin to GPU 0 if you ever have multiple
|
||||||
|
# CUDA_VISIBLE_DEVICES: "0"
|
||||||
|
|
||||||
|
# GPU access (Docker Compose)
|
||||||
|
deploy:
|
||||||
|
resources:
|
||||||
|
reservations:
|
||||||
|
devices:
|
||||||
|
- driver: nvidia
|
||||||
|
count: 1
|
||||||
|
capabilities: [gpu]
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
ollama:
|
ollama:
|
||||||
|
|||||||
Reference in New Issue
Block a user