dask和cuda交互的实用程序
dask-cuda的Python项目详细描述
达斯克库达
改善DASK和启用CUDA之间互操作的各种实用程序 系统。
这个存储库是为dask和cuda实用程序设计的。 它是实验性的,不应该被依赖。
当前包括
LocalCUDACluster
:一个dask.distributed.LocalCluster
的子类 简化了单节点多GPU系统的部署。
示例
fromdask_cudaimportLocalCUDAClusterfromdask.distributedimportClientcluster=LocalCUDACluster()client=Client(cluster)