R 中 LInux 上的 mxnet 多核
mxnet multicore on LInux in R
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"
尽管如此，它在 core <- 1
时可以正常工作
这是解决方案。
首先设置环境变量
Sys.setenv(MXNET_CPU_WORKER_NTHREADS =核心);
根据cpu/gpu
使用ctx=mx.cpu()
它正在 Windows 和 Linux
上工作
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"
尽管如此，它在 core <- 1
时可以正常工作这是解决方案。
首先设置环境变量
Sys.setenv(MXNET_CPU_WORKER_NTHREADS =核心);
根据cpu/gpu
使用ctx=mx.cpu()它正在 Windows 和 Linux
上工作