mirror of
https://github.com/steveseguin/caption-local.git
synced 2026-10-09 17:49:30 +00:00
17 lines
393 B
YAML
17 lines
393 B
YAML
# Use together: docker compose -f compose.yaml -f compose.gpu.yaml up --build -d
|
|
services:
|
|
captions:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile.gpu
|
|
image: caption-local:gpu
|
|
environment:
|
|
CAPTION_DEVICE: cuda
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|