gpt4 book ai didi

Python Caffe cpu & gpu 模式同时

转载 作者:行者123 更新时间:2023-11-28 18:36:29 25 4
gpt4 key购买 nike

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

最佳答案

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

关于Python Caffe cpu & gpu 模式同时,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32021975/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com