快速入门
可以使用LocalCUDACluster或从命令行使用dask cuda worker来创建一个Dask-CUDA集群。
本地CUDA集群
要使用所有可用的GPU创建一个Dask-CUDA集群，并将Dask.distributed Client连接到它：
from dask_cuda import LocalCUDACluster
from dask.distributed import Client
cluster = LocalCUDACluster()
client = Client(cluster)
提示
请确保在使用dask_cuda.LocalCUDACluster的独立Python脚本中包含一个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")