Caffe/pyCaffe: set all GPUs

Question

is possible to set all GPUs for Caffe (especially pyCaffe)?

Something like:

caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all

Answer 1

Both forks have supported multi-GPU for a while now.

• BVLC/caffe got support for multi-GPU on 08/13/2015 (see commit, issue).
• NVIDIA/caffe got support for multi-GPU on 06/19/2015 (see release note).

You may be interested to know that there is a pretty serious outstanding issue with multi-GPU and PythonLayers (see issue, temporary fix).