How to use multiple GPUs for training?

ArtyZe / yolo_segmentation

image (semantic segmentation) instance segmentation by darknet or yolo

186 stars 44 forks source link

How to use multiple GPUs for training? #26

Open yes89929 opened 3 years ago

yes89929 commented 3 years ago

Hi. Thanks you for sharing good code.

Now I`m training your model. But It seems like using 1 GPU.

Can I use multiple GPUs by just using arguments on terminal?

need I modify some code?

JasonRuan5 commented 2 years ago

@yes89929 How do you make 1 GPU working? I got build error when I enable GPU and CUDNN (issue #27). It would be great to know how you made 1 GPU to work.

JasonRuan5 commented 2 years ago

@yes89929, Sorry, I meant issue #29.