跳转到主要内容

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 库(如 RAPIDSCuPy)是一个持续的工作。您可能希望阅读 blog.dask.org 了解更多相关信息。有关 Dask-CUDA 的更多信息也可以在 docs 中找到。

项目详细信息


发行历史 发行通知 | RSS 源

下载文件

下载适合您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分发

dask_cuda-24.8.2.tar.gz (1.4 kB 查看散列)

上传时间

构建分发

dask_cuda-24.8.2-py3-none-any.whl (129.6 kB 查看散列)

上传时间 Python 3

由以下支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面