快速入门

Dask-CUDA 集群可以使用 LocalCUDACluster 或命令行中的 dask cuda worker 创建。

LocalCUDACluster

要使用所有可用 GPU 创建 Dask-CUDA 集群并连接一个 Dask.distributed 客户端

from dask_cuda import LocalCUDACluster
from dask.distributed import Client

cluster = LocalCUDACluster()
client = Client(cluster)

提示

在独立的 Python 脚本中使用 dask_cuda.LocalCUDACluster 时，请确保包含 if __name__ == "__main__": 块。更多详细信息请参见 独立的 Python 脚本。

dask cuda worker

要从命令行创建一个等效的集群，Dask-CUDA 工作进程必须连接到使用 dask scheduler 启动的调度器。

$ dask scheduler
distributed.scheduler - INFO -   Scheduler at:  tcp://127.0.0.1:8786

$ dask cuda worker 127.0.0.1:8786

要将客户端连接到此集群

from dask.distributed import Client

client = Client("127.0.0.1:8786")