从客户端或调度程序中关闭dask工作者

Question

在API中，有一种方法可以重新启动所有工作程序并完全关闭客户端，但是我看不到在保持客户端不变的情况下停止所有工作程序的方法。有没有办法解决我找不到的功能，或者它是不存在的功能？

Answer 1

这似乎是一个不存在的功能，但是使用当前代码仍然可以实现。您可以使用run_on_scheduler与调度程序本身的方法进行交互。

workers = list(c.scheduler_info()['workers'])
c.run_on_scheduler(lambda dask_scheduler=None: 
    dask_scheduler.retire_workers(workers, close_workers=True))

c客户在哪里，我们调用retire_workers优雅地要求每个工作人员退出。

可能还有其他方法可以实现这一目标。请注意，在这种情况下，调度程序仍在运行-从问题中尚不清楚是否需要这样做。