linux - Caffe の CPU コア使用量を制限しますか?

Question

Caffe の CPU コア使用率を制限する方法はありますか? 私の例では、Xeon E5-2699 を使用しており、Caffe を 9 コアの使用に制限したいので、CPU の 50% を使用します。トレーニングのほとんどは GPU で行われますが、それまでの間、他の開発に取り組みたいと考えています。これを行う簡単な方法はありますか？CentOS の実行

Answer 1

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 は /sys/fs/cgroup/cpu/fivecpus/cpu.cfs_period_us の値の 5 倍であることに注意してください。