是否可以为Caffe(尤其是pyCaffe)设置所有GPU?
就像是:
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all
这两个分支现在都支持多 GPU 一段时间了。
- BVLC/caffe 于 2015 年 8 月 13 日获得了对多 GPU 的支持(请参阅commit https://github.com/BVLC/caffe/commit/bb0a90e8aabb0e1ffd5732439591814d10d7fd45, issue https://github.com/BVLC/caffe/pull/2903).
- NVIDIA/caffe 于 2015 年 6 月 19 日获得了对多 GPU 的支持(请参阅发行公告 https://github.com/NVIDIA/caffe/releases/tag/v0.12.0).
您可能有兴趣知道多 GPU 和 PythonLayers 存在一个非常严重的突出问题(请参阅issue https://github.com/BVLC/caffe/issues/2936, 临时修复 https://github.com/BVLC/caffe/pull/3032).
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)