Question

有没有办法限制Caffe的CPU核心使用？对于我的实例，我有一个Xeon E5-2699，我想限制Caffe使用9个内核，所以50％的CPU。大多数培训都是在GPU上完成的，我还有其他一些我想在此期间开展的工作。是否有捷径可寻？运行CentOS

Answer 1

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

Answer 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>

限制Caffe CPU核心使用？

2 个答案: