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