是关于mxnet中的multiplecpu的概念

时间:2016-05-31 16:17:35

标签: mxnet

正如我们所知,mxnet中有关于multiplegpu的概念,当在命令中指定-gpu时。如果我们不指定gpu,它将在cpu上运行它。如果有多个cpus运行?是否可以指定多个cpu?

2 个答案:

答案 0 :(得分:1)

您可以使用多个具有以下代码的CPU(R版本,在python中非常相似):

devices = lapply(1:2, function(i) {
    mx.cpu(i)
})

和往常一样训练网络。此外,如果您有MKL库,系统会自动计算所有核心。

答案 1 :(得分:0)

有几种方法可以看这个。

如果使用良好的BLAS库编译MXNet,那些数学运算将使用所有可用的CPU核心。

此外,您可以通过环境变量MXNET_CPU_WORKER_NTHREADS指定多少CPU工作线程。见http://mxnet.io/how_to/env_var.html