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)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. Parallelizing GPU libraries like RAPIDS and CuPy with Dask is an ongoing effort. You may wish to read about this effort at blog.dask.org for more information..