英伟达 UMAP 支持多 GPU,处理 870GB 数据仅需 8 分钟
leland_mcinnes · x · 2026-08-20
NVIDIA cuML 和 cuVS 库更新,为 UMAP 算法引入了多 GPU 支持。新实现通过平衡聚类分区、独立计算局部 kNN 图并合并,避免了全对全通信。在 MIRACL 和 Wiki 数据集上的测试显示,使用 8 张 H100 GPU 可比 CPU 运行时提速最高 74 倍,使处理数百 GB 级工作负载成为可能,同时保持了嵌入质量。
所属事件:英伟达 UMAP 实现多 GPU 加速,870GB 数据仅需 8 分钟(2 条相关)→
「Infra」频道最新
- FT 称 AI 算力债券面临「达尔文时刻」,CoreWeave 债价暴跌 — firstadopter · 2026-10-16
- 手机当界面、家里服务器跑模型:一套本地 AI 双工作流 — EAccelerate_42 · 2026-10-16
- 「数据中心就是银行」:CoreWeave 风波引出算力监管类比 — arthurcolle · 2026-10-16
- DOT-MoE:用最优传输把稠密模型改造成稀疏 MoE — burkov · 2026-10-16
- 斯坦福 CS336:7B 模型权重仅 14GB,训练却要 84GB 起步 — stanfordnlp · 2026-10-16
- NVIDIA NVFP4 新压缩技巧:利用行内尺度相近性省 45-48% 存储且位精确 — TheZachMueller · 2026-10-16