search

clovaai / deep-text-recognition-benchmark

Text recognition (optical character recognition) with deep learning methods, ICCV 2019
Apache License 2.0
3.75k stars 1.1k forks source link

Multi GPUs core dumped #202

Open ghost opened 4 years ago

ghost commented 4 years ago

Hi, when I try to train model with multi GPUs(CUDA_VISIBLE_DEVICES=0,1,2 python3 train.py …), I got an error like 'core dumped'，while train with single GPU is fine

best

ZenFSheng commented 4 years ago

Hi, when I try to train model with multi GPUs(CUDA_VISIBLE_DEVICES=0,1,2 python3 train.py …), I got an error like 'core dumped'，while train with single GPU is fine

best

I think you have to set "num_workers" to 0 in dataloader, if you wanna train with multi gpus.

freedom9393 commented 4 years ago

Hi, when I try to train model with multi GPUs(CUDA_VISIBLE_DEVICES=0,1,2 python3 train.py …), I got an error like 'core dumped'，while train with single GPU is fine best

I think you have to set "num_workers" to 0 in dataloader, if you wanna train with multi gpus.

Still didn't work for me ...