限制芹菜中同时执行的特定任务的数量

Question

我有两个任务taskA和taskB，它们使用GPU运行。它们可以同时在GPU上运行。但是，如果芹菜决定启动一个额外的实例，那么我将崩溃(GPU内存不足(。

因此，我希望能够控制它(即，只允许taskA+taskB，或taskB+任务栏，或者任务栏+任务栏A(，但防止例如askA+

jobA我该如何控制？

Answer 1

您可以通过使用--concurrency参数来限制同时执行的任务的数量。如果两个任务都使用同一个工作者，请将其设置为2；如果有两个单独的工作者，请设置为1。