Exemplos de setup_local_dask_cluster em Python

Exemplo n.º 1

Arquivo: test_rmat.py Projeto: rapidsai/cugraph

def setup_module():
    global _cluster
    global _client
    global _visible_devices
    if not _is_single_gpu:
        (_cluster, _client) = setup_local_dask_cluster(p2p=True)
        _visible_devices = _client.scheduler_info()['workers']

Exemplo n.º 2

def client_connection():
    (cluster, client) = setup_local_dask_cluster(p2p=True)
    yield client
    teardown_local_dask_cluster(cluster, client)

Exemplo n.º 3

def setup_module():
    global _cluster
    global _client
    if not _is_single_gpu:
        (_cluster, _client) = setup_local_dask_cluster(p2p=True)