限制Caffe CPU核心使用率?

有没有办法限制Caffe的CPU核心使用率? 对于我的例子,我有一个Xeon E5-2699,我想限制Caffe使用9核心,所以CPU的50%。 大部分的培训都是在GPU上完成的,我还有一些其他的发展愿意在这个时候进行。 是否有捷径可寻? 运行CentOS

我不确定caffe是如何被调用的,但是您可以使用cgroups来限制centos6和之后的进程的cpu使用情况。 将cpu使用限制为5个逻辑cpus的示例:

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倍

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

在CentOS上安装taskset

 sudo yum install util-linux 

然后调用您的训练/推理脚本并将其限制为使用核心0至8(9核心):

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