search

mjq11302010044 / RRPN_pytorch

RRPN in pytorch 1.0 ----- Arbitrary-Oriented Scene Text Detection via Rotation Proposals
MIT License
262 stars 56 forks source link

Some questions of Multi-GPU training #42

Open Pluto1314 opened 4 years ago

Pluto1314 commented 4 years ago

when i use : CUDA_VISIBLE_DEVICES="3,4" python -m torch.distributed.launch --nproc_per_node 2 train_net.py i meet some trouble, like this: RuntimeError: CUDA error: an illegal memory access was encountered terminate called without an active exception How can I solve it，thanks a lot!

ShaneYS commented 4 years ago

have you solved problem about multi-gpu training?