NVIDIA 开源 cuPhoton 加速科学图像分析
来自 NVIDIA Developer · 期号 2026-10-07
摘要
天文望远镜和 X 射线光源等高通量仪器产生的图像数据,CPU 流水线来不及处理。NVIDIA 开源了 cuPhoton,一个 CUDA-X 工具包,让图像数据从传感器读出到分类全程留在 GPU 上,不再反复在 CPU 和 GPU 之间搬运。 以 Vera C. Rubin 天文台为例:LSSTCam 每 39 秒拍一张 32 亿像素的南天照片,一夜产生 20 TB 数据和 1000 万个候选天体,必须在 60 到 120 秒内完成分类。cuPhoton 把加载加速了最高 14900 倍,信号处理加速了最高 14550 倍,把几个月的计算时间压缩到几分钟。 工具包包含五个模块:xDataReader 直接把 FITS 格式加载到 GPU 内存、xRep 做图像对齐、xPois 做 PSF 匹配与减除、xFit 做偶极子拟合、xScan 做候选体审查。支持多 GPU 多节点扩展,在 Grace Blackwell 和 Vera Rubin 系统上运行。