HumanSignal/label-studio-ml-backend

MMDetection CUDA spawn

eflydata opened this issue · 0 comments

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