R中的LInux上的mxnet多核

Question

虽然在Linux上抛出错误

，但Follwoing在Windows上工作正常

core <- 5
ctx <- lapply(c(1:core), function(i) {mx.cpu(i)})

模型函数是“mx.model.FeedForward.create”

The error is "Error in (function (symbol, ctx, grad.req = "null", ...) : Need more shape information to decide the shapes of arguments"

虽然核心＆lt; - 1

时它可以正常工作

Answer 1

这是解决方案。

首先设置环境变量

Sys.setenv（MXNET_CPU_WORKER_NTHREADS =核心）;

使用ctx = mx.cpu（）取决于cpu / gpu

它适用于Windows和Linux