cuda - 多 GPU 上的 cuGraph

Question

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

Answer 1

cuGraph 通过利用 Dask 支持多 GPU。我鼓励您阅读 Dask cuGraph文档，该文档显示了使用 PageRank 的示例。

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

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