Microsoft TensorFlow-DirectML 正式版リリース: WSL での GPU による機械学習の高速化

IT Homeは9月12日、Microsoftが最近、Windows Subsystem for Linux（WSL）上でGPUを使用して機械学習（ML）モデルのトレーニングを加速できるツールであるTensorFlow-DirectMLの正式版をリリースしたと報じた。

名前が示すように、TensorFlow-DirectML は、Google の TensorFlow 機械学習プラットフォームを Microsoft の DirectML 機械学習 API に適合させ、Win10/WSL 上で機械学習モデルをトレーニングするときに GPU アクセラレーションをサポートするために使用されます。

IT Home は、ユーザーがpip install tensorflow-directmlと入力して Python 環境にインストールできることを知りました (Python 3.8 以降は現在サポートされていません)。これにより、さまざまな DirectX 12 互換ハードウェア上で複雑な機械学習モデルのトレーニングと推論がサポートされます。

Win10 のハードウェアおよびソフトウェア要件:

• Windows 10 1709、64 ビット (ビルド 16299 以上)

• Python x86-64 3.5、3.6、または 3.7;

サポートされている以下の GPU のいずれか:

• AMD Radeon R5/R7/R9 2xx シリーズ以降

• Intel HD Graphics 5xx 以降;

• NVIDIA GeForce GTX 9xx シリーズ GPU 以降

WSL のハードウェアおよびソフトウェア要件:

• Windows 10 Insider Preview、64 ビット (ビルド 20150 以上)。

• Python x86-64 3.5、3.6、または 3.7;

サポートされている以下の GPU のいずれか:

• AMD Radeon R5/R7/R9 2xx シリーズ以降、および 20.20.01.05 WSL ドライバー。

• Intel HD Graphics 6xx 以降、および 28.20.100.8322 WSL ドライバー。

• NVIDIA GeForce GTX 9xx シリーズ以降、および 460.20 WSL ドライバー。

具体的な使い方については、TensorFlow-DirectML の GitHub ページを確認してください。