RAPIDS

データの加工や表形式データの深層学習ではPandasが便利だが,大規模になると遅い.ちゃんとしたGPU付きのマシンがあるなら,GPUで加速したいところだ.いいパッケージがないかと探していたところ,RAPIDSというのを見つけた.

Pandas,Scikit Learn, NetworkXと似たAPIをGPU上で動かすことができるようだ.Dockerでインストールできるようなので,大学のマシンに入れてもらって試してもらおう.

特に,需要予測を大量の製品・顧客の対に対して行うには,通常の計算では間に合わない.RAPIDSにはHolt-Winterの季節変動を考慮した指数平滑法の例題がのっていたが,8倍くらいにはなるようだ.

以下の記事にもあるように,fastaiも対応していて,推奨システムを15倍速くできたそうだ.

大規模な推奨システムの場合には,速度よりもメモリの方が心配だが,応用によっては使えるかもしれない.



この記事が気に入ったらサポートをしてみませんか?