parallel-processing - Caffe/pyCaffe : set all GPUs

Question

是否可以为 Caffe(尤其是 pyCaffe)设置所有 GPU？

就像是:

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

Answer 1

两个分支现在都支持多 GPU 一段时间了。

BVLC/caffe 于 2015 年 8 月 13 日获得了对多 GPU 的支持(参见 commit、issue)。

NVIDIA/caffe 于 2015 年 6 月 19 日获得了对多 GPU 的支持(参见 release note)。

您可能有兴趣知道多 GPU 和 PythonLayers 存在一个非常严重的突出问题(请参阅 issue 、 temporary fix )。