Quickstart

A Dask-CUDA cluster can be created using either LocalCUDACluster or dask-cuda-worker from the command line.

LocalCUDACluster

To create a Dask-CUDA cluster using all available GPUs and connect a Dask.distributed Client to it:

from dask_cuda import LocalCUDACluster
from dask.distributed import Client

cluster = LocalCUDACluster()
client = Client(cluster)

dask-cuda-worker

To create an equivalent cluster from the command line, Dask-CUDA workers must be connected to a scheduler started with dask-scheduler:

$ dask-scheduler
distributed.scheduler - INFO -   Scheduler at:  tcp://127.0.0.1:8786

$ dask-cuda-worker 127.0.0.1:8786

To connect a client to this cluster:

from dask.distributed import Client

client = Client("127.0.0.1:8786")