🕐

Jupyter のセル内のコードの実行速度計測のためのメモ書き

2021/03/20に公開

(2021/3/21 追記)GPUを使うときの留意点のリンク先を追加

import time

start = time.time()

# ここにコードを入れる

print(f"elapsed_time:{round((time.time() - start), 1)}[sec]")

出力例

elapsed_time:6.8[sec]

小数点以下2位で四捨五入しています。

いつも下記サイトを参考にさせていただいているのですが、自分が使いやすいように修正していたので自分用のメモとして残します。

https://qiita.com/fantm21/items/3dc7fbf4e935311488bc

(2021/3/21 追記)また、GPUを使うときはCPU上の処理とGPU上の処理にずれが生じるので注意が必要とのことです。
詳しくは下記リンク先をご参照ください。

https://www.mattari-benkyo-note.com/2021/03/21/pytorch-cuda-time-measurement/

以上になります、最後までお読みいただきありがとうございました。

Discussion