CUDA : out of memory (2 vs 0 ) error

MhLiao / TextBoxes_plusplus

TextBoxes++: A Single-Shot Oriented Scene Text Detector

Other

953 stars 279 forks source link

CUDA : out of memory (2 vs 0 ) error #88

Closed rahil-patel closed 5 years ago

rahil-patel commented 5 years ago

Hi, I am getting a CUDA out of memory error while running the model, My Graphics specs are: NVIDIA GeForce GTX 1060, 6GB GDDR5 video memory, can you suggest how i can rectifyt this issue

outstandingking commented 5 years ago

hi @rahil-patel , it because that the author use two gpus, and u only get one. The config file path is 、、、、、、TextBoxes_plusplus/examples/text/modelConfig.py line 304 , change it to

#line 304 TextBoxes_plusplus/examples/text/modelConfig.py   

gpus= '0'