ray - 退出前等待所有任务完成的标准方法

Question

我想知道 - 是否有一种直接的方法可以在退出之前等待所有任务完成运行而不跟踪所有 ObjectID(和 get()ing他们)？用例是当我启动 @remote 以保存输出时，例如，在不需要返回结果的情况下。如果我必须存储这些 future ，这只是额外的东西来跟踪。

Answer 1

目前没有标准的方法来阻止所有任务完成。

有一些解决方法可以使用。

1. 跟踪列表 object_ids 中的所有对象 ID，然后调用 ray.get(object_ids) 或 ray.wait(object_ids , num_returns=len(object_ids))。

2. 只要一些资源被使用就循环。

   import time
   
   while (ray.global_state.cluster_resources() !=
          ray.global_state.available_resources()):
       time.sleep(1)
   

   上面的代码会一直循环，直到检测到当前没有任务在执行。然而，这不是万无一失的方法。可能有一段时间没有任务在运行，但调度程序任务即将开始运行。