NVIDIA的cuPyNumeric 25.03全面开源，增强PIP和HDF5支持
realtime news Apr 23, 2025 13:39
NVIDIA已将cuPyNumeric 25.03开源，通过PIP安装和原生HDF5支持增强其可访问性，促进多GPU计算的透明性和协作。
NVIDIA宣布其cuPyNumeric 25.03库现已完全开源，这标志着其开发的一个重要里程碑。根据NVIDIA，此次更新引入了强大的新功能，包括支持PIP安装和原生HDF5 IO。
完全开源过渡
在最新版本中，NVIDIA已将cuPyNumeric的整个堆栈开源，包括支持其的Legate框架和运行时层，并采用Apache 2许可证。这一过渡彰显了NVIDIA促进开发社区透明性、可重复性和协作的承诺。开源的性质使贡献者可以无障碍地探索、审计和增强系统。
PIP安装支持
以前只能通过conda安装的cuPyNumeric现在可以通过PIP安装，大大简化了设置过程。这一增强功能便于工作流、虚拟环境和CI管道的集成。PyPI上的cuPyNumeric包支持多节点和多排名功能，支持单节点多GPU和多GPU多节点集群。
原生HDF5 IO支持
cuPyNumeric 25.03的另一个显著特点是通过GPU直接存储支持HDF5，这优化了大数据集的处理。此功能确保了高效的IO操作，这对于高性能计算和数据密集型应用至关重要。用户现在可以以更高的性能和可移植性管理复杂的数据结构。
安装和使用
安装过程已简化为一个简单的PIP命令：pip install nvidia-cupynumeric。此更新捆绑了除MPI之外的所有主要依赖项，而这些依赖项可以通过PyPI轻松解决。NVIDIA提供了在SLURM集群上设置和运行cuPyNumeric的详细指南，强调了其在多节点和多排名环境中的易用性。
有关更多详细信息和探索cuPyNumeric 25.03的全面功能，NVIDIA鼓励用户查看官方发布说明并通过GitHub仓库为持续开发做出贡献。
Image source: Shutterstock