Caffe / pyCaffe：设置所有GPU

作者: 蜡笔小辛
发布时间: 2024-12-28 02:00:17 (1月前)
转自：

4 条回复

0#
回复此人
林老爷的日常 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> AFAIK Caffe目前不支持多gpu培训。计划将来发布。参见讨论 <a href="https://github.com/BVLC/caffe/issues/876" rel="noreferrer"> 这里 </A> 。 <BR/> 这好像是 <a href="https://github.com/NVIDIA/caffe" rel="noreferrer"> NVIDIA的caffe分支 </A> 有这个功能。看到问题 <a href="https://github.com/NVIDIA/DIGITS/issues/92" rel="noreferrer"> 这里 </A> 。 </p> </DIV>

编辑
1#
回复此人
布偶的表弟派大星丶 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 看起来Caffe现在开始支持C ++接口的多GPU培训。查看文档 <a href="http://caffe.berkeleyvision.org/tutorial/interfaces.html" rel="nofollow"> CommandLine接口 </A> 。 </p> <pre> <code> # 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 </code> </pre> </DIV>

编辑
2#
回复此人
甲基蓝 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 两种叉子现在已经支持多GPU了一段时间。 </p> <UL> <LI> 2015年8月13日，BVLC / caffe获得了对多GPU的支持（参见 <a href="https://github.com/BVLC/caffe/commit/bb0a90e8aabb0e1ffd5732439591814d10d7fd45"> 承诺 </A> ， <a href="https://github.com/BVLC/caffe/pull/2903"> 问题 </A> ）。 </LI> <LI> 2015年6月19日，NVIDIA / caffe获得了对多GPU的支持（参见 <a href="https://github.com/NVIDIA/caffe/releases/tag/v0.12.0"> 发行公告 </A> ）。 </LI> </UL> <P> 您可能有兴趣知道多GPU和PythonLayers存在一个相当严重的突出问题（请参阅 <a href="https://github.com/BVLC/caffe/issues/2936"> 问题 </A> ， <a href="https://github.com/BVLC/caffe/pull/3032"> 临时修复 </A> ）。 </p> </DIV>

编辑

登录后才能参与评论