search

rcaceiro / node-yolo

MIT License
18 stars 4 forks source link

Is there a way to select a specific GPU? #8

Open moeiscool opened 5 years ago

moeiscool commented 5 years ago

Hi there :) I recently got a GRID M40 running in my rig but I have to use a GT 1030 to get video ports.

So here is the problem : When I run yolo it uses the GT 1030 and not one of the GPUs provided by my M40.

What's worse is that my GT 1030 is actually plugged in to my second slot so I'm confused why yolo chose my last GPU :P

Is there a parameter i can specify during initialization or installation to use a particular card? I've attached a screenshot below.

Thanks for your effort on this! so many people love your module :) myself included!

unknown (5)

freakstatic commented 5 years ago

Hi @moeiscool Sorry for the late response We will try to add support to select the GPU as soon as we can

We are happy to see people using the module! :)

moeiscool commented 5 years ago

yay! thank you! I'm unfortunately a pure novice when it comes to C/C++ but I would like to help! Is there a patreon or donation link i can contribute to?

rcaceiro commented 5 years ago

Hi, @moeiscool No we haven't a donation link or any patreon associated to us, yet.

rcaceiro commented 5 years ago

Hello again, @moeiscool we already quick analyze your problem and the darknet seems to allow that feature only on training fase, this means that, or we and you open an issue on yolo's github or you most wait a little to allow us to change the structure, because we are Msc students and for now, the time is pretty limited, because Msc projects that we have. That change already was scheduled by me as a new feature but the lack of time not allow to implement this. Sorry for the inconvenience.