Various utilities to improve deployment and management of Dask workers on CUDA-enabled systems.
This library is experimental, and its API is subject to change at any time without notice.
from dask_cuda import LocalCUDACluster
from dask.distributed import Client
cluster = LocalCUDACluster()
client = Client(cluster)Documentation is available here.
This library does not automatically convert your Dask code to run on GPUs. It only helps with deployment and management of Dask workers in multi-GPU systems.