3
是否可以在CPU和GPU模式下运行Caffe?我有几个Caffe模型,但是我的GPU资源是有限的,所以我不能把所有模型放到GPU内存中。我想用例如3种GPU模式和2种CPU模式,但set_mode_cpu()
和set_mode_gpu()
命令只是切换整个库的模式。Python Caffe cpu和GPU模式同时运行
是否可以在CPU和GPU模式下运行Caffe?我有几个Caffe模型,但是我的GPU资源是有限的,所以我不能把所有模型放到GPU内存中。我想用例如3种GPU模式和2种CPU模式,但set_mode_cpu()
和set_mode_gpu()
命令只是切换整个库的模式。Python Caffe cpu和GPU模式同时运行
在我看来,你可以为每个任务编写一个python脚本。 在每个脚本中,您可以选择是使用CPU还是GPU(和GPU设备)。 然后您可以同时运行这些多个脚本。 但是在我的经验中,在一张GPU卡上执行多个任务时会严重降低速度。祝你好运!