最近，我正在阅读cuGraph的代码。我注意到有人提到Louvain和Katz算法支持多GPU。然而，当我阅读Louvain的C++代码时，我找不到与多GPU相关的代码。具体地，根据先前的帖子，多GPU可以通过调用cudaSetDevice来实现。然而，我在Louvain的代码中找不到这个函数。我遗漏了什么吗？
cuGraph通过利用Dask支持多GPU。我鼓励您阅读Dask-cuGraph文档，其中显示了使用PageRank的示例。
对于Louvain示例，我建议查看cugraph.dask.louvain函数的docstring。
为了完整起见，cuGraph正在使用RAFT来管理底层NCCL和UCX通信。