Scikit-Learn 1.8引入 Array API,支持 PyTorch 与 CuPy 张量的原生 GPU 加速

Scikit-Learn 1.8.0 更新引入了实验性的 Array API 支持。这意味着 CuPy 数组或 PyTorch 张量现在可以直接在 Scikit-Learn 的部分组件中直接使用了,且计算过程能保留在 GPU 上。

 

https://avoid.overfit.cn/post/ab7e632896364fc3b4b9fdc9d17884e3

posted @ 2025-12-18 21:28  deephub  阅读(3)  评论(0)    收藏  举报