NumPyとCuPyの処理速度比較

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

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

Stock環境構築手順_20240611.xlsx

6.09 MB

ファイルダウンロードについて

ダウンロード

■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の方が遅くなっとるがな。
労多くして功少なしとはこのことか。
まあ、やってみたかっただけだし。
( ﾟдﾟ)､ﾍﾟｯ