众所周知,mxnet中有multiplegpu的概念,当在命令中指定-gpu时。如果我们不指定gpu,它将在cpu上运行。它运行多少个cpu?可以指定多个cpu吗?
问问题
556 次
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 回答