使用NVIDIA nvmath-python大规模运行高性能核心数学计算
Run High-Performance Core Math at Scale with NVIDIA nvmath-python
摘要
NVIDIA 发布 nvmath-python,一个连接 Python 科学计算社区与 NVIDIA CUDA-X 数学库的库,为 Python 用户提供 CUDA-X 性能,用于常见数学运算且不干扰现有工作流程。根据 API 不同,运算可在 CPU、支持 CUDA 的 GPU 或分布式多 GPU、多节点系统上运行。
NVIDIA nvmath-python 是一个旨在弥合 Python 科学计算社区与 NVIDIA CUDA-X 数学库之间差距的库。它为 Python 用户提供了访问 CUDA-X 性能的能力,用于常见数学运算,而不会干扰现有工作流程。根据 API 的不同,运算可以在 CPU、支持 CUDA 的 GPU 或分布式多 GPU、多节点系统上运行。来源
译自 NVIDIA · Developer 博客 · 录于 二〇二六年七月三十一日