Python Caffe cpu& gpu模式同时进行

时间:2015-08-15 05:48:01

标签: python caffe

是否可以在CPU和GPU模式下运行Caffe?我有几个Caffe模型,但我的GPU资源有限,因此我无法将所有模型都放入GPU内存中。我想用例如3种型号采用GPU模式,2种型号采用CPU模式,但set_mode_cpu()set_mode_gpu()命令只需切换整个库的模式。

1 个答案:

答案 0 :(得分:1)

在我看来,你可以为每个任务编写一个python脚本。 在每个脚本中,您可以选择使用CPU还是GPU(和GPU设备)。 然后,您可以同时运行这些多个脚本。 但是根据我的经验,在一张GPU卡上运行多个任务会严重降低速度。祝你好运!