1

我有一个功能可以有效地生成图像并存储到磁盘中。该函数没有参数:

def generate_and_save():
    pass # generate and store image

我需要生成大量图像(比如 100k),所以我选择了 Dask。阅读手册后,我整理了一个创建分布式(本地)客户端并使用多个进程执行任务的函数,如下所示:

from dask.distributed import Client as DaskClient

def generate_images(how_many=10000, processes=6):
    # start Dask distributed client with 1 thread per process
    client = DaskClient(n_workers=processes, threads_per_worker=1)
    # submit future functions to cluster
    futures = []
    for i in range(how_many): 
        futures.append(client.submit(generate_and_save, pure=False))
    # execute and compute results (synchronous / blocking!)
    results = client.gather(futures)
    print(len(results))
    # stop & release client
    client.close()

generate_images(50000)

如您所见,“期货”在 for 循环中提交给服务器并存储在一个简单的列表中。问题是:在这种情况下,是否有更有效的方式来添加和执行期货?例如,并行化提交过程本身?

4

1 回答 1

2

没有。这看起来很不错。我不希望开销花费太长时间,每个任务可能不到 1 毫秒,所以 10 秒

如果此开销很长,那么您可能需要阅读此文档部分:https ://docs.dask.org/en/latest/best-practices.html#avoid-very-large-graphs

于 2019-09-21T00:57:43.450 回答