彼得·张
2025年4月23日 13:20
NVIDIA已开源cuPyNumeric 25.03,增强了其可访问性,支持PIP安装和原生HDF5,促进了多GPU计算中的透明性和协作。
NVIDIA宣布其cuPyNumeric 25.03库现已完全开源,这标志着其发展的一个重要里程碑。根据<a href="https://developer.nvid
完全开源转型
在这次最新发布中,NVIDIA 已经将整个 cuPyNumeric 堆栈开源,包括支撑它的 Legate 框架和运行时层,采用 Apache 2 许可证。这一转型强调了 NVIDIA 在开发社区中促进透明性、可重复性和协作的承诺。开源的特性使得贡献者能够无障碍地探索、审计和增强系统。
PIP 安装支持
之前只能通过 conda 安装的 cuPyNumeric 现在可以使用 PIP 安装,大大简化了设置过程。这一增强使得在工作流、虚拟环境和 CI 管道中的集成更加容易。PyPI 上的 cuPyNumeric 包支持多节点和多排名,能够支持单节点多 GPU 和多 GPU 多节点集群。
原生 HDF5 IO 支持
cuPyNumeric 25.03 的另一个显著特点是通过 GPU 直接存储对 HDF5 的原生支持,优化了大数据集的处理。此功能确保了高效的 IO 操作,对于高性能计算和数据密集型应用至关重要。用户现在可以以更好的性能和可移植性管理复杂的数据结构。
安装和使用
安装过程已被简化,包含一个简单的 PI
P命令:pip install nvidia-cupynumeric
。此更新捆绑了所有主要依赖项,除了MPI,后者可以通过PyPI轻松解决。NVIDIA提供了关于在SLURM集群上设置和运行cuPyNumeric的详细指南,强调了在多节点和多进程环境中的易用性。
有关更多详细信息以及探索cuPyNumeric 25.03的全面功能,NVIDIA鼓励用户查看官方发布说明,并通过GitHub存储库为持续开发做出贡献。
图片来源:Shutterstock