Utilities for Dask and CUDA interactions
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.
What this is not
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. Additional information about Dask-CUDA can also be found in the docs.
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size dask_cuda-0.18.0-py3-none-any.whl (67.6 kB)||File type Wheel||Python version py3||Upload date||Hashes View|
|Filename, size dask-cuda-0.18.0.tar.gz (66.4 kB)||File type Source||Python version None||Upload date||Hashes View|
Hashes for dask_cuda-0.18.0-py3-none-any.whl