Caffe/pyCaffe:设置所有GPU [英] Caffe/pyCaffe: set all GPUs
本文介绍了Caffe/pyCaffe:设置所有GPU的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
是否可以为Caffe(尤其是pyCaffe)设置所有GPU?
is possible to set all GPUs for Caffe (especially pyCaffe)?
类似的东西:
caffe train -solver examples/mnist/lenet_solver.prototxt -gpu all
推荐答案
AFAIK Caffe目前不支持多GPU训练.计划将来发布.在此处中查看讨论.
似乎 NVIDIA的caffe分支具有此功能.请在此处看到问题.
AFAIK Caffe is not supporting multi gpu training at the moment. It is planned for future release. See a discussion here.
It seems like NVIDIA's branch of caffe has this functionality. See the issue here.
这篇关于Caffe/pyCaffe:设置所有GPU的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文