跳转到主要内容

pylibcugraph - Python绑定libcugraph cuGraph C/C++/CUDA库

项目描述


cuGraph


RAPIDS cuGraph是一个单代码库,代表了一组关注GPU加速图分析的软件包,包括对属性图、远程(图作为服务)操作和图神经网络(GNN)的支持。cuGraph支持创建和操作图,然后执行可扩展的快速图算法。


新闻

新功能! nx-cugraph,一个提供零代码更改的GPU加速的NetworkX后端。

> pip install nx-cugraph-cu11 --extra-index-url https://pypi.nvidia.com
> export NETWORKX_AUTOMATIC_BACKENDS=cugraph

就是这样。NetworkX现在利用cuGraph进行加速图算法。


目录




Stack

RAPIDS cuGraph 是一套GPU加速的图算法和服务集合。在Python层,cuGraph在GPU DataFrames上操作,从而允许在cuDF中的ETL任务和cuML中的机器学习任务之间无缝传递数据。熟悉Python的数据科学家可以很快掌握cuGraph如何与cuDF类似Pandas的API集成。同样,熟悉NetworkX的用户也会很快识别出cuGraph中提供的类似NetworkX的API,目的是使现有代码以最小的努力移植到RAPIDS。为了简化集成,cuGraph还支持Pandas DataFrameNetworkX 图对象和其他几种格式中的数据。

虽然cugraph Python API提供了数据科学家易于使用且熟悉的接口,与其他RAPIDS库的流程保持一致,但某些用例需要访问更底层的图论概念。对于这些用户,我们提供了一个额外的Python API,称为pylibcugraph,旨在需要与cuGraph在Python层进行更紧密集成且依赖较少的应用程序。熟悉C/C++/CUDA和图结构的使用者可以访问libcugraph和libcugraph_c,以实现Python之外的低级集成。

注意:要确保您使用的是最新的README.md稳定版本,请检查您是否在最新的分支上。

以下是一个示例Python代码片段,它加载图数据并计算PageRank

import cudf
import cugraph

# read data into a cuDF DataFrame using read_csv
gdf = cudf.read_csv("graph_data.csv", names=["src", "dst"], dtype=["int32", "int32"])

# We now have data as edge pairs
# create a Graph using the source (src) and destination (dst) vertex pairs
G = cugraph.Graph()
G.from_cudf_edgelist(gdf, source='src', destination='dst')

# Let's now get the PageRank score of each vertex by calling cugraph.pagerank
df_page = cugraph.pagerank(G)

# Let's look at the top 10 PageRank Score
df_page.sort_values('pagerank', ascending=False).head(10)

为什么cuGraph不支持方法级联


使用cuGraph的项目

(按字母顺序排列)

(如果您想将项目添加到此列表,请提出问题)



开放GPU数据科学

RAPIDS开源软件库系列旨在使数据科学和数据分析管道完全在GPU上执行。它依赖于NVIDIA® CUDA®原语进行底层计算优化,并通过用户友好的Python接口暴露GPU并行性和高带宽内存速度。

有关更多项目详情,请参阅rapids.ai



GPU上的Apache Arrow

Apache Arrow的GPU版本是一个通用的API,它可以在运行在GPU上的进程之间有效地交换表格数据。在GPU上的端到端计算避免了从GPU上不必要的复制和转换数据,减少了在人工智能工作负载中常见的高性能分析的计算时间和成本。正如其名所示,cuDF在GPU上使用Apache Arrow列式数据格式。目前,Apache Arrow功能的一部分得到支持。

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分布

pylibcugraph_cu12-24.8.0.tar.gz (3.8 kB 查看哈希值)

上传时间

由以下支持