2025年12月18日
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/ab7e 阅读全文
posted @ 2025-12-18 21:28 deephub 阅读(7) 评论(0) 推荐(0)