Caffe/pyCaffe：设置所有GPU

Question

可以为Caffe（特别是pyCaffe）设置所有GPU？

喜欢的东西：

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

Answer 1

AFAIK来自Caffe是不支持目前的多GPU训练。它计划在未来发布。查看讨论here。
好像NVIDIA's branch of caffe有这个功能。请参阅问题here。

Answer 2

看来Caffe现在开始支持针对C++接口的多GPU的培训。 请参阅CommandLine interface上的文档。

# train on GPUs 0 & 1 (doubling the batch size) 
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu 0,1 
# train on all GPUs (multiplying batch size by number of devices) 
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all 

Answer 3

这两款叉子现在都支持多GPU。

• BVLC /朱古力得到了支持多GPU在2015年8月13日（见commit，issue）。
• NVIDIA/caffe在06/19/2015获得了对多GPU的支持（请参阅release note）。

你可能有兴趣知道，还有多GPU和PythonLayers一个相当严重的突出问题（见issue，temporary fix）。