HumanSignal / label-studio-ml-backend

Configs and boilerplates for Label Studio's Machine Learning backend
Apache License 2.0
587 stars 262 forks source link

MMDetection CUDA spawn #560

Open eflydata opened 5 months ago

eflydata commented 5 months ago

docker runtime tip: RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method - DEVICE=cuda:0