多 GPU 上的 cuGraph

cuGraph on Multi-GPU

最近在看cuGraph的代码。我注意到有人提到 Louvain 和 Katz 算法支持多 GPU。然而,当我阅读Louvain, I cannot find code that is related to multi-GPU. Specifically, according to a prior post的C++代码时,可以通过调用cudaSetDevice来实现多GPU。但是,我在 Louvain 的代码中找不到这个函数。我错过了什么吗?

cuGraph 通过利用 Dask 支持多 GPU。我鼓励您阅读 Dask cuGraph documentation,其中显示了使用 PageRank 的示例。

对于 Louvain 示例,我建议查看 cugraph.dask.louvain 函数的文档字符串。

为了完整性,cuGraph 在幕后使用 RAFT 来管理底层 NCCL 和 UCX 通信。