文章/答案/技术大牛

发布

社区首页 >问答首页 >Caffe/pyCaffe:设置所有GPU

问Caffe/pyCaffe:设置所有GPU
EN

Stack Overflow用户

提问于 2015-11-15 23:23:54

回答 3查看 5.4K关注 0票数 4

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

类似于：

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

parallel-processing

neural-network

gpu

deep-learning

caffe

回答 3

Stack Overflow用户

回答已采纳

发布于 2015-11-16 07:03:02

AFAIK Caffe目前不支持多gpu训练。计划在未来发布。参见讨论这里。

[医]仙人掌科似乎具有此功能。请参阅问题这里。

票数 6

Stack Overflow用户

发布于 2016-01-22 18:25:58

两个叉子已经支持多GPU一段时间了。

2015年8月13日，BVLC/caffe获得了对多GPU的支持(见提交，问题)。
2015年6月19日，NVIDIA/caffe获得了多GPU的支持(见释放说明)。

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

票数 7

Stack Overflow用户

发布于 2015-11-18 15:28:21

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

# 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

票数 4

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/33726333

复制

相似问题

问Caffe/pyCaffe:设置所有GPU
EN

回答 3

Stack Overflow用户

Stack Overflow用户

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Caffe/pyCaffe:设置所有GPUEN

回答 3

Stack Overflow用户

Stack Overflow用户

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Caffe/pyCaffe:设置所有GPU
EN