-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdocker-compose.yml
More file actions
37 lines (37 loc) · 1.32 KB
/
docker-compose.yml
File metadata and controls
37 lines (37 loc) · 1.32 KB
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
27
28
29
30
31
32
33
34
35
36
37
# PipeRouter compute backend. Run on the GPU workstation:
# docker compose up --build -d
# The Omniverse extension (on the host) writes voxel grids into
# /dev/shm/piperouter/<session>/ and calls the solver at http://localhost:8000.
services:
solver:
build: ./services/solver
image: piperouter-solver
container_name: piperouter-solver
ports:
- "${PIPEROUTER_PORT:-8000}:8000"
environment:
- PIPEROUTER_GRID_DIR=/dev/shm/piperouter
- WIRE_TYPES_PATH=/app/wire_types.json
# Build the routing graph on the GPU (cupy) and hand it to cuGraph zero-copy —
# ~2x faster route_all, no CPU->GPU edge transfer. Falls back to numpy if cupy is
# unavailable. (Smoothing stays CPU; set PIPEROUTER_GPU_SMOOTH=1 to also GPU that.)
- PIPEROUTER_GPU_BUILD=1
volumes:
# shared grid handoff: same tmpfs path on host and container
- /dev/shm/piperouter:/dev/shm/piperouter
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
healthcheck:
test:
- CMD
- python3
- -c
- "import urllib.request,sys; sys.exit(0 if urllib.request.urlopen('http://localhost:8000/health').status==200 else 1)"
interval: 10s
timeout: 5s
retries: 10