Windows 10 und WSL: GPU-beschleunigtes ML-Training mit TensorFlow-DirectML von Microsoft
Vor über einem Jahr kündigte Microsoft an, dass es mit Hardware-Anbietern zusammenarbeitet, um GPU-beschleunigtes Training von Machine-Learning-Modellen (ML) auf Windows Subsystem for Linux (WSL) anzubieten. Eine Preview dafür begann im Juni 2020. Das Ergebnis dieser Initiative ist TensorFlow-DirectML, ein Fork von TensorFlow, der DirectML nutzt, um herstellerübergreifende Hardwarebeschleunigung für das Training von ML-Modellen auf Windows 10 und WSL zu bieten. Heute hat dieses Open-Source-Projekt auf GitHub die Vorschau verlassen und ist allgemein verfügbar.
Microsoft hat betont, dass die Nutzung von TensorFlow-DirectML recht einfach ist, da es in einer Python-Umgebung mit dem Befehl "pip install tensorflow-directml" installiert werden kann. Danach wird es automatisch in Ihre bestehenden Skripte integriert, die Modelle trainieren. Der Redmonder Tech-Riese sagt, dass er direkt mit Studenten und Fachleuten zusammengearbeitet hat, um ihre Skripte abzudecken und Operatoren wie Batch-Normalisierung und Faltung zu optimieren. Dieser Prozess umfasste auch die Verbesserung der GPU-Planung und der Speicherverwaltungsmechanismen.
Microsoft hat mit Anbietern wie Nvidia, AMD und Intel zusammengearbeitet, um sicherzustellen, dass Windows 10 und WSL reibungslos funktionieren und das Training auf DirectX 12-fähigen GPUs beschleunigt wird. Derzeit sind die Systemanforderungen für TensorFlow-DirectML wie folgt:
Windows-Subsystem für Linux
- Windows 10 Insider Preview, 64-Bit (Build 20150 oder höher)
- Python x86-64 3.5, 3.6 oder 3.7
Eine der folgenden unterstützten GPUs:
- NVIDIA GeForce GTX 9xx Serie GPU oder neuer, und 460.20 WSL-Treiber
- AMD Radeon R5/R7/R9 2xx-Serie oder neuer und WSL-Treiber 20.20.01.05
- Intel HD Graphics 6xx oder neuer, und WSL-Treiber 28.20.100.8322
Anmerkung: Python 3.8 oder neuer wird derzeit nicht unterstützt. Um die offiziellen PyPi-Pakete zu verwenden, ist der CPython-Interpreter erforderlich.
Windows 10
- Windows 10 Version 1709, 64-bit (Build 16299 oder höher)
- Python x86-64 3.5, 3.6, oder 3.7
Eine der folgenden unterstützten GPUs:
- NVIDIA GeForce GTX 9xx Serie oder neuere GPUs
- AMD Radeon R5/R7/R9 2xx-Serie oder neuer
- Intel HD Graphics 5xx oder neuer
Anmerkung: Python 3.8 oder neuer wird derzeit nicht unterstützt. Um die offiziellen PyPi-Pakete zu verwenden, wird der CPython-Interpreter benötigt. NumPy 1.19.4 erfordert KB4598291, um unter Windows ordnungsgemäß zu funktionieren.
Microsoft hat zugesichert, das Projekt mit gezielter Operator-Unterstützung weiter zu optimieren. Sie können die Open-Source-Initiative auf GitHub hier nachlesen.