米MicrosoftのWindows AIチームは、機械学習モデルをトレーニングするためのPyTorchのバックエンドとして使えるようにすべく、畳み込みニューラルネットワーク（CNNS）のサポートを提供する、PyTorch-DirectMLパッケージのプレビューリリース提供を、10月21日（現地時間）に開始した。
今回、発表されたプレビューリリースは、WindowsおよびWindows Subsystem for Linux（WSL）上のDirectX12 GPUによって、PyTorchの高速機械学習トレーニングを実現するための、最初のステップとなる。
PyTorch-DirectMLパッケージでは、DirectML APIとPyTorch Tensorプリミティブを呼び出す「DML」デバイスを導入することで、DirectMLがPyTorchフレームワークと統合される。DirectMLオペレータへの呼び出しのオーバーヘッドは最小限であり、DirectMLバックエンドは既存の他のPyTorchバックエンドと同じように機能する。
なお、このハードウェアアクセラレーションによるPyTorchのトレーニングは、AMD、Intel、NVIDIAとの共同設計によって実現されている。
同パッケージは、pipまたはPyPIを通じてインストールできる。
この記事は参考になりましたか？
- この記事の著者
-
CodeZine編集部（コードジンヘンシュウブ）
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です
