有没有办法限制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>