#7 · 主分类: 经典机器学习框架
cuml
NVIDIA cuML:GPU加速的机器学习
项目最后更新:08/28/26
GitHub Stars
5.3K
Forks数量
658
贡献者数量
206
许可证
Apache-2.0
收录理由
当团队的表格型工作负载在CPU上遇到性能瓶颈,又不想推倒重写现有代码时,cuML是一个很实际的选择。它的GPU原生估计器沿用了熟悉的scikit-learn fit-predict-transform模式,聚类、回归、分类和降维这些任务都能在CUDA上跑,而你依然使用已经掌握的API。更值得关注的是cuml.accel模块:把它指向现有的scikit-learn、UMAP或HDBSCAN脚本,支持的操作会自动迁移到GPU执行,遇到无法加速的估计器或配置时则回退到CPU,整个流程不会中断。这为加速遗留管道提供了一条低风险的路径。此外,分布式API还能应对超出单张GPU容量的数据集,让扩展性不再成为瓶颈。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。