0

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

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

谢谢您的回答!

4

1 回答 1

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 参数

希望这可以帮助!

于 2017-02-28T21:18:22.563 回答