File size: 417 Bytes
3df5819 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | version: "3.8"
services:
api:
build: .
ports:
- "8000:8000"
volumes:
- ./configs:/app/configs
- ./data:/app/data
- ./checkpoints:/app/checkpoints
environment:
- CUDA_VISIBLE_DEVICES=0
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
restart: unless-stopped
|