使用mkldnn编译mxnet CPP软件包以支持无环境变量的单线程执行

我们正在寻找一种在类似Linux的系统上通过单个线程使用mxnet CPP库的方法。 根据mxnet文档和社区经验,我们发现的唯一方法是设置环境变量:

> ifelse(df$date == 2018,2018,as.Date(df$date,origin="1899-12-30"))
[1]    NA    NA 17896 17896 17896 17896  2018    NA 17896 17896 17896 17896

但是这种方式是有问题的,并非总是可以配置环境变量。

深入了解mxnet代码 我们发现有一些预处理器定义和CMake标志可以提供帮助

export MXNET_ENGINE_TYPE=NaiveEngine
export OMP_NUM_THREADS=1

问题是:

1)是否可以完全在单线程模式下编译mxnet?

2)如果有可能,我们可以做多少最少的更改来编译单线程mxnet lib?

3)还有其他解决方案可确保mxnet神经网络模型的单线程执行吗?

任何帮助将不胜感激。

zhouhsmp3 回答:使用mkldnn编译mxnet CPP软件包以支持无环境变量的单线程执行

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3153809.html

大家都在问