espnet3.parallel.parallel.make_local_gpu_cluster

Less than 1 minute

espnet3.parallel.parallel.make_local_gpu_cluster(n_workers: int, options: dict) → Client

Create a Dask LocalCUDACluster using available GPUs.

This requires dask_cuda package.

Parameters:
- n_workers (int) – Number of Dask workers (must not exceed number of GPUs).
- options (dict) – Additional options for the LocalCUDACluster.
Returns: Dask client connected to the GPU cluster.
Return type: Client