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