在dask中为LocalCluster设置内存参数的正确方法

Question

我试过下面的代码，

from dask.distributed import Client, LocalCluster
worker_kwargs = {
    'memory_limit': '2G',
    'memory_target_fraction': 0.6,
    'memory_spill_fraction': 0.7,
    'memory_pause_fraction': 0.8,
    'memory_terminate_fraction': 0.95,
}
cluster = LocalCluster(ip='0.0.0.0', n_workers=8, **worker_kwargs)

但它不起作用

...
TypeError: __init__() got an unexpected keyword argument 'memory_target_fraction'

什么是正确的打电话方式？ LocalCluster似乎也不关心~/.config/dask/distributed.yaml。

我读过这些帖子：

1. https://github.com/dask/distributed/issues/2456
2. https://github.com/dask/distributed/issues/1521

Answer 1

似乎是版本问题。将dask更新到1.2.0可以解决问题。

from dask.distributed import Client, LocalCluster
worker_kwargs = {
    'memory_limit': '2G',
    'memory_target_fraction': 0.6,
    'memory_spill_fraction': 0.7,
    'memory_pause_fraction': 0.8,
    # 'memory_terminate_fraction': 0.95,
}
cluster = LocalCluster(ip='0.0.0.0', n_workers=8, **worker_kwargs)

作品