AMD active enfin PyTorch sur Windows et Linux

AMD tient ses promesses et franchi une étape importante pour les développeurs et chercheurs en intelligence artificielle. Le géant déploie une préversion publique de ROCm 6.4.4, une mise à jour qui permet à PyTorch de fonctionner nativement aussi bien sous Windows que sous Linux. De plus la compatibilité est assurée  sur une large gamme de GPU grand public.

PyTorch arrive sur Radeon RX 9000, RX 7000 et Ryzen AI

Cette version est majeure car elle développe le ROCm sur les piles Windows et Linux en apportant la prise en charge native de PyTorch aux PCs équipés d’une Radeon RX 9000 (RDNA 4), Radeon RX 7000 (RDNA 3) ou encore de certains APU Ryzen AI 300 “Strix” et Ryzen AI MAX “Strix Halo”. Comme nous l’avons souligné en introduction, AMD concrétise une promesse faite lors du dernier Computex, celle de rendre ROCm multiplateforme et plus accessible aux développeurs.

Jusqu’à présent utiliser PyTorch avec une carte Radeon sous Windows relevait du véritable casse-tête. L’opération s’appuyait sur un double démarrage, des contournements compliqués ou encore un recours au cloud. Tout ceci est de l’histoire ancienne. Les développeurs peuvent désormais installer et tester et de prototyper des modèles IA localement, tout en donnant leur retour pour affiner les performances et la compatibilité.

Une étape seulement

Attention tout de même AMD présente cette préversion comme une base de travail, et non comme une solution définitive. Le projet va donc évoluer et s’améliorer mais le but est connu. ROCm va se déployer au-delà des centres de données, où il équipe déjà les accélérateurs Instinct et les serveurs EPYC, pour trouver son chemin vers les PCs des créateurs, des chercheurs et des développeurs grand public.

La feuille de route prévoit un cycle de mises à jour plus rapides, avec des optimisations continues et une meilleure intégration Windows dans les prochaines versions, notamment avec ROCm 7.0 destiné aux environnements professionnels.