linux - 限制 Caffe CPU 核心使用率？

Question

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

Answer 1

要限制任何进程使用的核心，请使用taskset.

taskset在 CentOS 上安装：

sudo yum install util-linux

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

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

Answer 2

我不确定如何调用 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 倍