caffe-segnet icon indicating copy to clipboard operation
caffe-segnet copied to clipboard

Can you use multiple GPUs?

Open KevinLucidyne opened this issue 8 years ago • 1 comments

Looking through the code, I'm mainly seeing only a single GPU being used. Is it possible to use multiple GPUs?

KevinLucidyne avatar Jun 24 '16 20:06 KevinLucidyne

yes, form the command line, you would do something like this caffe train -solver solver.prototxt -gpu 0,1 where 0 and 1 are the IDs of the GPUs you want to use. You can also just replace 0,1 with the word 'all' and Caffe will make use of all of the GPUs in your system.

jorgyz avatar Aug 17 '16 19:08 jorgyz