#7 · 主分类: 经典机器学习框架

cuml

cuda gpu machine-learning machine-learning-algorithms nvidia

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容量的数据集,让扩展性不再成为瓶颈。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目