NVlabs/Deep_Object_Pose

Do I have to multi gpus??

Opened this issue · 1 comments

I'm trying to train dope with script at train2.
But my computer has only one GPU(RTX-3050).
Do I must use multi gpus??
Is there any way to train dope with one GPU?

You do not need multi gpus, the code uses the distributed package from pytorch, so you need to start it with the distributed packages starting lines. I think with a 3050 you might need to reduce quite a bit the batchsize.