米Microsoftは、WindowsおよびWindows Subsystem for Linux(WSL)上のDirectX12 GPUにて、PyTorchモデルをトレーニングするためのハードウェアアクセラレーションバックエンド「PyTorch-DirectML」の、2番目のプレビューを2月25日(現地時間)にリリースした。
2番目のプレビューでは、パフォーマンスを大幅に向上するとともに、コンピュータビジョンモデルのカバレッジを拡大している。
なお、PyTorch-DirectMLはPython 3.6/3.7/3.8で動作し、PyPIからダウンロードできる。
この記事は参考になりましたか?
- この記事の著者
-
CodeZine編集部(コードジンヘンシュウブ)
CodeZineは、株式会社翔泳社が運営するソフトウェア開発者向けのWebメディアです。「デベロッパーの成長と課題解決に貢献するメディア」をコンセプトに、現場で役立つ最新情報を日々お届けします。
※プロフィールは、執筆時点、または直近の記事の寄稿時点での内容です