見出し画像

NumPyとCuPyの処理速度比較

新しくゲーミングPCを買いました。以前予告していたとおりNumPyをCuPyに置き換えてGPUで計算させたら処理速度は速くなるのか検証しました。

■CuPyを使用するための環境構築
過去にアップした環境構築手順にCuPyのシートを追加しました。

■PCスペック
PCのスペックは以下となっております。
メーカー:マウスコンピューター
モデル:G-Tune DG-I7G7S
CPU:インテル(R) Core(TM) i7-14700F
メモリ:32GB
グラフィックス:NVIDIA GeForce RTX 4070 SUPER / 12GB

■検証方法
「output/trade/myself/new_high.py」でバックテストを実行して検証しました。new_high.pyでは「signals/indicator.py」のhighest()を呼び出しています。highest()でnumpy.max()を使用しています。これをcupy.amax()に置き換えて比較しました。

バックテストの条件
条件①:過去20年間の株価データ - 20銘柄
条件②:過去5年間の株価データ - 3840銘柄(※)
※2024年4月30日時点の東証上場全銘柄

ソース修正の都合で以下の順序で実行しました。
条件①:numpy.max() → cupy.amax()
条件②:cupy.amax() → numpy.max()

■検証結果
条件①:過去20年間の株価データ - 20銘柄
★numpy.max()
開始:2024-06-11 10:05:07.122281
終了:2024-06-11 10:05:44.450205
約37秒

★cupy.amax()
開始:2024-06-11 10:07:11.721151
終了:2024-06-11 10:08:05.865140
約54秒

条件②:過去5年間の株価データ - 3840銘柄
★numpy.max()
開始:2024-06-11 10:55:14.680588
終了:2024-06-11 11:24:25.062948
約29分

★cupy.amax()
開始:2024-06-11 10:12:32.370486
終了:2024-06-11 10:51:10.272485
約39分

cupyの方が遅くなっとるがな。
労多くして功少なしとはこのことか。
まあ、やってみたかっただけだし。
( ゚д゚)、ペッ


いいなと思ったら応援しよう!