米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は、株式会社翔泳社が運営する開発者のための情報メディアです。日々の開発に取り組むエンジニアやテクノロジーを学びたい方に向けて、プログラミングやAI活用、開発ツール、エンジニアの学びとキャリアに関する記事をお届けしています。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です
この記事は参考になりましたか？
この記事をシェア
