Python setup_local_dask_cluster Examples

Programming Language: Python

Namespace/Package Name: cugraph.dask.common.mg_utils

Method/Function: setup_local_dask_cluster

Examples at hotexamples.com: 3

Python setup_local_dask_cluster - 3 examples found. These are the top rated real world Python examples of cugraph.dask.common.mg_utils.setup_local_dask_cluster extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

File: test_rmat.py Project: 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']

Example #2

Show file

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

Example #3

Show file

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