linux – 限制Caffe CPU核心使用？

发布时间：2020-12-13 23:20:10 所属栏目：Linux 来源：网络整理

导读：有没有办法限制Caffe的CPU核心使用？对于我的实例,我有一个Xeon E5-2699,我想限制Caffe使用9个内核,所以50％的CPU.大多数培训都是在GPU上完成的,我还有其他一些我想在此期间开展的工作.是否有捷径可寻？运行CentOS 解决方法我不确定如何调用caffe,但您可以

有没有办法限制Caffe的CPU核心使用？对于我的实例,我有一个Xeon E5-2699,我想限制Caffe使用9个内核,所以50％的CPU.大多数培训都是在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倍

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!