使用分布式环境(多节点)时,我可以在mxnet中设置工作负载吗?

时间:2017-02-27 12:53:02

标签: mxnet

我想问一下,当我使用mxnet使用分布式计算环境时,是否可以设置不同的工作负载。我读了一些分布式GPU的教程。

但是我想使用分布式节点(CPU)环境,我想为它们设置不同的工作负载。我能这样做吗?如果是,那么我能得到一些关于它的例子吗?

感谢您的回答!

1 个答案:

答案 0 :(得分:1)

是的,它是受支持的。检查此链接,该链接显示您可以根据要分配工作负载的GPU或CPU指定work_load_list。 http://mxnet.io/how_to/multi_devices.html#advanced-usage

此外,您应该检查python API引用(http://mxnet.io/api/python/model.html#mxnet.model.FeedForward)。可以在执行model.Feedforward.fit(__)

时设置work_load_list参数

希望这有帮助!