cuGraph - 基于 GPU 的图形分析
Apache
跨平台
C/C++
软件简介
RAPIDS cuGraph库是一组图形分析,用于处理GPU数据帧中的数据 -
请参阅cuDF。
cuGraph旨在提供类似NetworkX的API,这对数据科学家来说很熟悉,因此他们现在可以更轻松地构建GPU加速的工作流程。
示例代码:
import cugraph
# assuming that data has been loaded into a cuDF (using read_csv) Dataframe
# create a Graph using the source (src) and destination (dst) vertex pairs the GDF
G = cugraph.Graph()
G.add_edge_list(gdf["src"], gdf["dst"])
# Call cugraph.pagerank to get the pagerank scores
gdf_page = cugraph.pagerank(G)
for i in range(len(gdf_page)):
print("vertex " + str(gdf_page['vertex'][i]) +
" PageRank is " + str(gdf_page['pagerank'][i]))