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]))