Dask 和 CUDA 交互的实用工具
项目描述
Dask CUDA
提供各种工具,以改善在具有 CUDA 功能的系统上部署和管理 Dask 工作进程。
此库是实验性的,其 API 可能随时更改,恕不另行通知。
示例
from dask_cuda import LocalCUDACluster
from dask.distributed import Client
cluster = LocalCUDACluster()
client = Client(cluster)
文档可在 此处 查找。
这不是什么
此库不会自动将您的 Dask 代码转换为在 GPU 上运行。
它仅帮助在多 GPU 系统中部署和管理 Dask 工作进程。与 Dask 并行化 GPU 库(如 RAPIDS 和 CuPy)是一个持续的工作。您可能希望阅读 blog.dask.org 了解更多相关信息。有关 Dask-CUDA 的更多信息也可以在 docs 中找到。
项目详细信息
下载文件
下载适合您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。
源分发
dask_cuda-24.8.2.tar.gz (1.4 kB 查看散列)
构建分发
dask_cuda-24.8.2-py3-none-any.whl (129.6 kB 查看散列)