是否可以在某些工作列表上运行dask / distributed作业

时间:2017-03-16 15:55:37

标签: python distributed dask

我有一个包含40名工作人员的计算网络,但我只需要在某些主机上执行计算。是否可以限制或重建网络以仅使用指定的工作人员?

1 个答案:

答案 0 :(得分:0)

是的,您可以使用workers=关键字为client.submit,client.map,client.compute或client.persist指定特定的worker。

futures = client.map(func, args, workers=['worker-1', 'worker-2'])

参见http://distributed.readthedocs.io/en/latest/locality.html#user-control  了解更多选择

您还可以向工作人员注册GPU等资源,并将计算限制为这些资源。有关详细信息,请参阅http://distributed.readthedocs.io/en/latest/resources.html