PyTorchでGPUの計算時間を正しく計測する | まったり勉強ノート

今回の記事ではPyTorchでGPUで実行した関数の計算時間を正しく測定する方法とその後に詳しい説明をしていきます。はじめに仕事がらPyTorchで高速な学習方法をいろいろ調べることがよくあります。その際、blog記事などで、Pythonの time() を利用して計算時間を測定して「こんなに速くなりました!」という紹介記事を見かけ…