在Deep Learning" mxnet"中,限制核心数(cpu)

时间:2017-06-15 03:40:43

标签: deep-learning mxnet

命令" ctx = mx.cpu()"正在占用所有可用的CPU。如何限制只使用一定数量 - 比如8个核心中的6个

1 个答案:

答案 0 :(得分:1)

不幸的是 - 没有。即使cpu上下文将int作为input argument

def cpu(device_id=0):
    """Returns a CPU context.

根据官方文件:

 Parameters
 ----------
 device_id : int, optional
     The device id of the device. `device_id` is not needed for CPU.
     This is included to make interface compatible with GPU.

然而,理论上,由于device_id参数存在,因此将来可能会更改。但是现在MXNet占用了所有可用内核。