File size: 496 Bytes
694c514 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
services:
nemo-retriever-ocr:
build:
context: .
dockerfile: Dockerfile
deploy:
resources:
reservations:
devices:
- capabilities: [gpu]
working_dir: /workspace
volumes:
- .:/workspace:rw
- ${XDG_CACHE_HOME:-~/cache}:/root/.cache:rw
command: bash -lc "python example.py ocr-example-input-1.png --merge-level paragraph"
ipc: host
ulimits:
memlock:
soft: -1
hard: -1
stack: 6710886
|