Rapids引入GPU PURRARS流和统一的GNN API增强功能

NVIDIA宣布了其Rapids Suite的最新版本25.06，这是Python Data Science的CUDA-X库集。 Nvidia表示，该版本引入了几种旨在提高计算效率和数据处理能力的开创性功能。

Polars GPU发动机增强功能

新版本为Polars GPU引擎带来了重大更新，该发动机最初于2024年9月启动。一个关键功能是实验流opecutor，它允许通过数据分配和并行处理大于可用VRAM的数据集上的执行。这种开发对于加速非常大的数据集的分析操作至关重要，从数百千兆字节缩放到trabytes。此外，该更新引入了混乱机制，以促进设备之间的数据重新分布，并支持多GPU执行。

另一个增强功能包括对滚动聚合和扩展的列操作功能的支持，这对于时间序列数据分析特别有益。现在，GPU引擎还支持DateTime列操作的更广泛的表达式，例如 .strftime() 和 .cast_time_unit()。

图形神经网络（GNNS）的统一API

将WholeGraph集成到NVIDIA的CUGRAPH-PYG中，导致了统一的API的创建，该API加速了GNN的提取。此API允许用户无缝地从单个GPU过渡到多GPU或多节点工作流，而无需修改其脚本。熟悉的 torchrun Pytorch的命令用于管理过程，从而促进Pytorch用户的易用性。

零代码更改ML增强功能

Rapids 25.06版本通过在CUML库中包括支持向量机（SVM），扩展了机器学习的零代码变化加速度。这允许使用SVM的现有Scikit-Learn工作流从GPU加速度中受益，而无需进行任何代码修改。该更新提高了与Scikit-Learn的兼容性，增强了参数验证和错误处理。

其他平台和兼容性更新

该版本还包括对Rapids Memory Manager（RMM）的升级，该升级现在支持NVIDIA Blackwell GPU上的基于硬件的减压引擎。此功能有望改善IO密集型工作流程。此外，该平台现在支持Python 3.13，标志着支持CUDA 11的最后一个版本。

总体而言，Rapids 25.06发布为数据科学家和开发人员带来了重大进步，重点是增强性能和易于使用GPU加速数据处理任务。

图像来源：Shutterstock