R 中 LLinux 上的 mxnet 多核

Question

Follwoing 在 Windows 上运行良好，但在 Linux 上抛出错误

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"

虽然，当核心

Answer 1

这是解决方案。

先设置环境变量

Sys.setenv(MXNET_CPU_WORKER_NTHREADS = core);

根据 cpu/gpu 使用 ctx= mx.cpu()

它可以在 Windows 和 Linux 上运行