CodeZine(コードジン)

特集ページ一覧

PyTorch向けの新しいパフォーマンスツール「PyTorch Profiler」がリリース

  • LINEで送る
  • このエントリーをはてなブックマークに追加
2021/03/29 08:00

 オープンソースのPython向け機械学習ライブラリPyTorchの開発チームは、PyTorchのための新たなパフォーマンスデバッグプロファイラ「PyTorch Profiler」を、3月25日(現地時間)に発表した。

 PyTorch Profilerは、米Microsoftと米Facebookとのコラボレーションの一環として開発された、大規模な深層学習モデルにおいて正確で効率的なパフォーマンス分析とトラブルシューティングを可能にするオープンソースツール。

 PyTorch固有の操作コンテキストを持たないものの、GPUハードウェアレベルの情報を提供するパフォーマンスデバッグツールと、詳細なGPUハードウェアレベルの情報をキャプチャせず、視覚化のサポートを提供できないものの、PyTorch操作に関する情報をキャプチャできるtorch.autograd.profilerの、両方のタイプの情報をまとめて、それらを相互に関連付け、モデルのボトルネックの自動検出を実行して、それを解決するための推奨事項を生成する。

 PyTorch Profilerからの情報はTensorBoardによって可視化され、新たなプロファイラAPIはPyTorchでネイティブにサポートされているので、追加パッケージをインストールすることなくモデルのプロファイリングが可能となり、TensorBoardにてすぐに結果を確認できる。

関連リンク

  • LINEで送る
  • このエントリーをはてなブックマークに追加

あなたにオススメ

All contents copyright © 2005-2021 Shoeisha Co., Ltd. All rights reserved. ver.1.5