Darknet: where to find GPU options in darknet if I want to use only second gpu?

[bit-scientist]

I am aware that if I add -gpu option to the training code I can train with multiple gpus, but what if I wanted to specifically choose the second one. Where in the darknet I can change that?

[AlexeyAB]

Just use such commands to use only second GPU:

darknet.exe detector train data/voc.data cfg/yolov3-voc.cfg /backup/yolov3-voc_1000.weights -gpus 1

Or such if you want to use GUP 0 and 3:

darknet.exe detector train data/voc.data cfg/yolov3-voc.cfg /backup/yolov3-voc_1000.weights -gpus 0,3

[bit-scientist]

thank you @AlexeyAB
I should've thought that before.

Also, could you please point the source code where I can see those options and others?

[vkage]

@git-sohib see workflow here , and modify according to your work.

[AlexeyAB]

@git-sohib

https://github.com/AlexeyAB/darknet/blob/e1e5abe19197f2bb56a9d320b31275db4706aa4d/src/detector.c#L1206-L1227

https://github.com/AlexeyAB/darknet/blob/e1e5abe19197f2bb56a9d320b31275db4706aa4d/src/detector.c#L1250

https://github.com/AlexeyAB/darknet/blob/e1e5abe19197f2bb56a9d320b31275db4706aa4d/src/darknet.c#L370