yolo_segmentation icon indicating copy to clipboard operation
yolo_segmentation copied to clipboard

How to use multiple GPUs for training?

Open yes89929 opened this issue 3 years ago • 2 comments

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?

OR

need I modify some code?

yes89929 avatar Jul 16 '21 04:07 yes89929

@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 avatar Dec 11 '21 02:12 JasonRuan5

@yes89929, Sorry, I meant issue #29.

JasonRuan5 avatar Dec 11 '21 02:12 JasonRuan5