Spaces:
Running
Running
File size: 509 Bytes
f90241e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
version: '3.8'
services:
dust3r-demo:
build:
context: ./files
dockerfile: cuda.Dockerfile
ports:
- "7860:7860"
environment:
- DEVICE=cuda
- MODEL=${MODEL:-DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth}
volumes:
- ./files/checkpoints:/dust3r/checkpoints
cap_add:
- IPC_LOCK
- SYS_RESOURCE
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
|