faststyle icon indicating copy to clipboard operation
faststyle copied to clipboard

Published 20 hours ago • verified publisher icon ghwatson

GPU scaling

Open petergerten opened this issue 8 years ago • 3 comments

Is there any way to train with multiple GPUs? The current scripts seem to only utilize the first GPU, altough TF 'sees' all of them.

petergerten avatar Mar 14 '17 03:03 petergerten

Unfortunately, I haven't yet played with the multi-tower training stuff as I only have a humble little 970 at the moment. Pull requests are welcomed!

ghwatson avatar Mar 14 '17 15:03 ghwatson

ok, thanks. I am quite new to TF though - will take a look.

petergerten avatar Mar 15 '17 07:03 petergerten

I'll leave this issue open in case someone implemented it in one of their forks and sees this.

ghwatson avatar Mar 15 '17 13:03 ghwatson