多GPU上的cuGraph

Question

最近，我正在阅读cuGraph的代码。我注意到，有人提到，Louvain和Katz算法支持多GPU。然而，当我阅读C++代码的卢万时，我找不到与多GPU相关的代码。具体来说，根据以前的帖子，多GPU可以通过调用cudaSetDevice来实现.然而，我在卢万的代码中找不到这个函数。我有遗漏什么吗？

Answer 1

cuGraph通过利用Dask来支持多GPU。我鼓励您阅读Dask cuGraph 文档，它展示了一个使用PageRank的示例。

对于Louvain示例，我建议查看cugraph.dask.louvain函数的docstring。

为了完整起见，cuGraph使用木筏来管理底层的NCCL和UCX通信。