限制Caffe CPU內核使用率？

有沒有辦法限制Caffe的CPU核心使用率？對於我的例子，我有一個Xeon E5-2699，我想限制Caffe使用9個內核，因此佔CPU的50％。大部分培訓都是在GPU上完成的，並且我還有一些其他發展願意在此期間進行。是否有捷徑可尋？運行CentOS限制Caffe CPU內核使用率？

來源

2016-04-25 Dimitri S

你使用的是什麼BLAS？ – Shai

我正在使用OpenBLAS –

openblas趨勢消耗CPU。它使用很多線程。你應該檢查如何限制openblas – Shai

我不確定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是5倍/sys/fs/cgroup/cpu/fivecpus/cpu.cfs_period_us

來源

2016-11-16 19:27:37

爲了限制芯中使用的值通過任何過程，使用taskset。

CentOS上安裝taskset：

sudo yum install util-linux

然後調用您的培訓/推理腳本，並限制其使用內核0-8（9芯）：

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

來源

2017-04-09 22:54:51 MohamedEzz

限制Caffe CPU內核使用率？

回答

相關問題