Scikit-learnが実験的にGPU対応していたので調査してみた！ - Qiita

はじめに みずほリサーチ＆テクノロジーズ株式会社の@fujineです。 いきなりですがAIエンジニアの皆さん、scikit-learnが実験的にGPUに対応していたこと、ご存知でしょうか？ scikit-learnは機械学習分野における古参パッケージの1つです。多様な機能を提供する一方、FAQにて「GPUに対応する予定はない（キリッ）」と公式宣言しており、scikit-learnが好きな自分としては「勿体無いなぁ」と常々感じていました。 そんな中、何気なくRelease Highlights for 1.2を読んでいたら以下文面を発見！しかも約半年前の2022年12月にリリースされてる… Experimental Array API support in LinearDiscriminantAnalysis Experimental support for the Array API sp

[misshiki]

“いきなりですがAIエンジニアの皆さん、scikit-learnが実験的にGPUに対応していたこと、ご存知でしょうか？”

[sh19910711]

"scikit-learn: ver1.2にて、GPUによる学習・推論が実験的に導入 / 学習データをnumpy.ndarrayオブジェクトからcupy.ndarrayオブジェクトに変換 / コンテキストマネージャーの中でfit()やtransform()、predict()を実行 / array_api_dispatch" 2023