限制Caffe CPU核心使用？

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>