オープンソースのPython向け機械学習ライブラリPyTorchの開発チームは、PyTorchのための新たなパフォーマンスデバッグプロファイラ「PyTorch Profiler」を、3月25日(現地時間)に発表した。
PyTorch Profilerは、米Microsoftと米Facebookとのコラボレーションの一環として開発された、大規模な深層学習モデルにおいて正確で効率的なパフォーマンス分析とトラブルシューティングを可能にするオープンソースツール。
PyTorch固有の操作コンテキストを持たないものの、GPUハードウェアレベルの情報を提供するパフォーマンスデバッグツールと、詳細なGPUハードウェアレベルの情報をキャプチャせず、視覚化のサポートを提供できないものの、PyTorch操作に関する情報をキャプチャできるtorch.autograd.profilerの、両方のタイプの情報をまとめて、それらを相互に関連付け、モデルのボトルネックの自動検出を実行して、それを解決するための推奨事項を生成する。
PyTorch Profilerからの情報はTensorBoardによって可視化され、新たなプロファイラAPIはPyTorchでネイティブにサポートされているので、追加パッケージをインストールすることなくモデルのプロファイリングが可能となり、TensorBoardにてすぐに結果を確認できる。
この記事は参考になりましたか?
- この記事の著者
-
CodeZine編集部(コードジンヘンシュウブ)
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です