Here is the code I am using to setup the client / cluster:
from dask_cuda import LocalCUDACluster from dask.distributed import Client cluster = LocalCUDACluster(