label-studio-ml-backend
label-studio-ml-backend copied to clipboard
MMDetection CUDA spawn
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