限制 Caffe CPU 核心使用率？答案

【问题标题】：Limit Caffe CPU core usage?限制 Caffe CPU 核心使用率？
【发布时间】：2016-08-19 04:05:08
【问题描述】：

有没有办法限制 Caffe 的 CPU 核心使用率？对于我的实例，我有一个 Xeon E5-2699，我想限制 Caffe 使用 9 个内核，即 50% 的 CPU。大部分训练都是在 GPU 上完成的，同时我还有一些其他的开发我想进行。是否有捷径可寻？运行 CentOS

【问题讨论】：

你用的是什么 blas？
我正在使用 OpenBLAS
openblas 趋向于消耗 CPU。它使用许多线程。你应该检查如何限制 openblas
谢谢。将其添加到我的搜索中就出现了这个线程：stackoverflow.com/questions/30195837/…

标签： linux python-2.7 caffe pycaffe

【解决方案1】：

我不确定如何调用 caffe，但您可以使用 cgroups 来限制 centos6 及更高版本进程的 CPU 使用率。将 cpu 使用限制为 5 个逻辑 cpu 的示例：

sudo mkdir -p /sys/fs/cgroup/cpu/fivecpus
sudo bash -c "echo 500000 > /sys/fs/cgroup/cpu/fivecpus/cpu.cfs_quota_us"
sudo bash -c "echo $$ > /sys/fs/cgroup/cpu/fivecpus/tasks"
your_command

请注意，500000 是 /sys/fs/cgroup/cpu/fivecpus/cpu.cfs_period_us 中值的 5 倍

【讨论】：

【解决方案2】：

要限制任何进程使用的内核，请使用taskset。

在 CentOS 上安装taskset：

sudo yum install util-linux

然后调用您的训练/推理脚本并将其限制为使用核心 0 到 8（9 个核心）：

taskset -c 0,1,2,3,4,5,6,7,8 <your_command>

【讨论】：