有没有关于mxnet中的多CPU的概念

Question

如我们所知，在mxnet中有关于multiplegpu的概念，当在命令中指定-gpu时。并且如果我们不指定gpu，它将在cpu上运行它。许多cpus做它运行？可以指定多个CPU吗？

Answer 1

您可以使用多个CPU，用下面的代码（R版本，在蟒蛇很相似）：

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

，列车网络和往常一样。另外，如果您有MKL库，系统会自动计算所有核心。

Answer 2

有几种方法来看待这个。

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

另外，您可以通过环境变量MXNET_CPU_WORKER_NTHREADS指定多少个CPU工作线程。请参阅http://mxnet.io/how_to/env_var.html