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']
def client_connection(): (cluster, client) = setup_local_dask_cluster(p2p=True) yield client teardown_local_dask_cluster(cluster, client)
def setup_module(): global _cluster global _client if not _is_single_gpu: (_cluster, _client) = setup_local_dask_cluster(p2p=True)