Caffe/pyCaffe:设置所有GPU

Question

可以为Caffe(特别是pyCaffe)设置所有GPU吗？

就像是:

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

Answer 1

两种叉子现在已经支持多GPU了一段时间.

• 2015年8月13日,BVLC/caffe获得了对多GPU的支持(请参阅commit,issue).
• 2015年6月19日,NVIDIA/caffe获得了对多GPU的支持(参见发行说明).

您可能有兴趣知道多GPU和PythonLayers存在一个相当严重的突出问题(请参阅问题,临时修复).

Answer 2

AFAIK Caffe目前不支持多gpu培训.计划将来发布.请参阅此处的讨论.
似乎NVIDIA的caffe分支具有此功能.在这里查看问题.