cuDF - GPU DataFrames

📢 cuDF现在可以作为无代码更改的加速器用于pandas！了解更多信息，请参阅此处！

cuDF（发音为“库迪夫”）是一个用于加载数据、连接、聚合、过滤以及其他操作数据的GPU DataFrame库。cuDF利用libcudf，一个快速的C++/CUDA DataFrame库和Apache Arrow列格式，提供GPU加速的pandas API。

您可以直接导入cudf并像使用pandas一样使用它

import cudf

tips_df = cudf.read_csv("https://github.com/plotly/datasets/raw/master/tips.csv")
tips_df["tip_percentage"] = tips_df["tip"] / tips_df["total_bill"] * 100

# display average tip by dining party size
print(tips_df.groupby("size").tip_percentage.mean())

或者，您可以使用cuDF作为pandas的无代码更改加速器，使用cudf.pandas。 cudf.pandas支持100%的pandas API，利用cuDF执行支持的操作，并在需要时回退到pandas

%load_ext cudf.pandas  # pandas operations now use the GPU!

import pandas as pd

tips_df = pd.read_csv("https://github.com/plotly/datasets/raw/master/tips.csv")
tips_df["tip_percentage"] = tips_df["tip"] / tips_df["total_bill"] * 100

# display average tip by dining party size
print(tips_df.groupby("size").tip_percentage.mean())

资源

• 立即尝试cudf.pandas：在Google Colab的免费GPU实例上探索cudf.pandas！
• 安装：cuDF和其他RAPIDS库的安装说明。
• cudf (Python) 文档
• libcudf (C++/CUDA) 文档
• RAPIDS 社区：获取帮助、贡献和协作。

查看RAPIDS 安装页面以获取安装cuDF和其他RAPIDS包的最新信息和命令。

安装

CUDA/GPU要求

• CUDA 11.2+
• NVIDIA驱动程序 450.80.02+
• Volta架构或更高（计算能力 >=7.0）

Pip

cuDF可以通过从NVIDIA Python包索引使用pip进行安装。请确保根据您环境中可用的CUDA主要版本选择适当的cuDF包。

对于CUDA 11.x

pip install --extra-index-url=https://pypi.nvidia.com cudf-cu11

对于CUDA 12.x

pip install --extra-index-url=https://pypi.nvidia.com cudf-cu12

Conda

cuDF可以通过conda（通过miniconda或从rapidsai频道的全Anaconda发行版）进行安装。

conda install -c rapidsai -c conda-forge -c nvidia \
    cudf=24.08 python=3.11 cuda-version=12.5

我们还提供从我们最新开发分支的HEAD构建的夜间Conda包。

注意：cuDF仅支持Linux，并且与Python 3.9及更高版本兼容。

查看RAPIDS 安装指南以获取更多操作系统和版本信息。

从源代码构建/安装

查看构建说明。

贡献

请参阅我们关于为cuDF做出贡献的指南。