darknet/docker-compose.yml

20 lines
367 B
YAML

version: '2'
services:
yolo-gpu:
build:
context: .
dockerfile: Dockerfile.gpu
image: yolo:gpu
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
yolo-cpu:
build:
context: .
dockerfile: Dockerfile.cpu
image: yolo:cpu