Au cours de la dernière année, Microsoft a développé une formation accélérée par GPU pour les modèles d'apprentissage automatique dans son service Windows Subsystem for Linux (WSL). Après la dernière année de préversion, Microsoft publie désormais le projet sous le nom de TensorFlow-DirectML.
Comme son nom l'indique, TensorFlow-DirectML est un fork qui inclut la plateforme d'apprentissage automatique TensorFlow de Google et l'API DirectML de Microsoft.
DirectML propose des outils d'accélération GPU pour les tâches d'apprentissage automatique. DirectML fonctionne sur le matériel et les pilotes des GPU DirectX 12 sur les puces AMD, Qualcomm, Nvidia et Intel. Avec TensorFlow-DirectML, Microsoft fournit une plate-forme d'accélération matérielle accessible à plusieurs fournisseurs pour former des modèles d'apprentissage automatique.
Annonce publicitaire
Microsoft affirme que le projet open source est désormais entièrement disponible via GitHub.
« Aujourd'hui, nous sommes ravis de sortir de la préversion et d'annoncer notre premier package grand public de TensorFlow-DirectML ! Nous vous recommandons d'utiliser TensorFlow-DirectML, que vous soyez un étudiant en formation ou un professionnel développant des modèles d'apprentissage automatique pour la production. Poursuivez votre lecture pour plus de détails sur la manière dont nous avons investi et vers où nous nous dirigeons ensuite.
Pour aider à publier la meilleure version de TensorFlow-DirectML, Microsoft a sollicité les commentaires des utilisateurs et a travaillé avec des partenaires tels qu'AMD, Intel et Nvidia. Microsoft indique que les exigences suivantes sont nécessaires pour exécuter le service.
Exigences
fenêtres 10
- Windows 10 version 1709, 64 bits (build 16299 ou supérieur)
- Python x86-64 3.5, 3.6 ou 3.7
- L'un des GPU pris en charge suivants :
- Série AMD Radeon R5/R7/R9 2xx ou plus récente
- Intel HD Graphics 5xx ou version ultérieure
- NVIDIA GeForce GTX série 9xx ou GPU plus récent
Note: Python 3.8 ou version ultérieure n'est actuellement pas pris en charge. Pour utiliser les packages PyPi officiels, l'interpréteur CPython est requis. NumPy 1.19.4 nécessite KB4598291 pour fonctionner correctement sous Windows.
Sous-système Windows pour Linux
- Windows 10 Insider Preview, 64 bits (Build 20150 ou version ultérieure)
- Python x86-64 3.5, 3.6 ou 3.7
- L'un des GPU pris en charge suivants :
- AMD Radeon R5/R7/R9 série 2xx ou version ultérieure et pilote WSL 20.20.01.05
- Intel HD Graphics 6xx ou version ultérieure et pilote WSL 28.20.100.8322
- NVIDIA GeForce GTX série 9xx ou GPU plus récent et pilote WSL 460.20
Note: Python 3.8 ou version ultérieure n'est actuellement pas pris en charge. Pour utiliser les packages PyPi officiels, l'interpréteur CPython est requis.
Le nouveau TensorFlow-DirectML est disponible sur GitHub ici.
Annonce publicitaire
