9 lines
194 B
YAML
9 lines
194 B
YAML
|
services:
|
||
|
|
main:
|
||
|
|
gpus: all
|
||
|
|
shm_size: "8gb"
|
||
|
|
environment:
|
||
|
|
NVIDIA_VISIBLE_DEVICES: all
|
||
|
|
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
||
|
|
OCR_DEVICE: "${OCR_DEVICE:-gpu:0}"
|