File size: 518 Bytes
0c5ad16 2fd019f 0c5ad16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
services:
detr_cpu:
profiles: [ "cpu" ]
build:
context: .
ports:
- 7000:7000
volumes:
- ./data:/app/data
detr_gpu:
build:
dockerfile: Dockerfile.nvidia
target: app
profiles: [ "gpu" ]
shm_size: '4gb' # pytorch dataloader crash
ports:
- 7000:7000
volumes:
- ./data:/app/data
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ] |