Caffe可以比较方便的使用多个GPU进行预测吗

比如有两个网络,一个在GPU0上做预测,另一个在GPU1上做预测
已邀请:

alex68 - 一般不扯淡~

赞同来自:

如果是两个网络对应两个进程,这肯定是没问题的,我原来在一个4卡K80服务器上试过,每个进程对应一个GPUn,跑了8个caffe进程,互相不影响。
如果是一个进程里对应两个网络,这就难说了,我用pycaffe试过,似乎不可以。不是很确定,希望其他朋友们确认一下。

tensorflow - CaffeCN社区Tensorflow主题站管理员 http://tensorflow.caffecn.cn

赞同来自:

哈 歪个楼 使用Tensorflow可以轻松设置和指定使用哪颗GPU : http://www.cnblogs.com/darkknightzh/p/6591923.html
 

要回复问题请先登录注册