3

众所周知,mxnet中有multiplegpu的概念,当在命令中指定-gpu时。如果我们不指定gpu,它将在cpu上运行。它运行多少个cpu?可以指定多个cpu吗?

4

2 回答 2

1

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

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

并且像往常一样训练网络。此外,如果您有MKL库,系统会自动使用所有内核进行计算。

于 2016-09-04T13:10:13.823 回答
0

有几种方法来看待这个。

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

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

于 2016-11-07T19:41:15.607 回答