我想问一下,当我使用mxnet使用分布式计算环境时,是否可以设置不同的工作负载。我读了一些分布式GPU的教程。
但是我想使用分布式节点(CPU)环境,我想为它们设置不同的工作负载。我能这样做吗?如果是,那么我能得到一些关于它的例子吗?
感谢您的回答!
答案 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参数希望这有帮助!