Is there a way to use CPU only?

multimodallearning / pytorch-mask-rcnn

Other

2.01k stars 557 forks source link

Is there a way to use CPU only? #63

Closed HanChangHun closed 3 years ago

HanChangHun commented 5 years ago

Hi. I am having problems because I do not have a GPU. I only have a laptop and I can not use another computer. I really want to run the demo. Can you tell me how?

Zteat commented 5 years ago

If you only need to run the demo.py ,you can change the code in the demo, line 36：GPU_COUNT=0.