AMD ROCm

ROCm est la réponse d’AMD à CUDA de NVIDIA. C’est un ensemble de logiciels qui permet aux programmes d’utiliser les processeurs graphiques AMD pour du calcul général, dont des pilotes, une chaîne de compilation, des bibliothèques de maths et de communication, et des outils de débogage et de profilage. Une grande partie est open source.

Pour le travail d’IA, la pièce clé est le soutien des frameworks. PyTorch, TensorFlow, JAX et d’autres tournent sur ROCm, et des outils d’inférence comme vLLM et beaucoup de bibliothèques de modèles ont des builds AMD. Les développeurs peuvent porter du code CUDA avec HIP, une couche de programmation qui ressemble à CUDA, et de grands déploiements cloud et de supercalculateurs utilisent les accélérateurs AMD Instinct avec ROCm.

ROCm est gratuit. Il a historiquement traîné CUDA en maturité et en soutien d’écosystème, donc la compatibilité avec des GPU et des bibliothèques précis varie et demande une vérification, mais l’écart s’est resserré. Les alternatives incluent NVIDIA CUDA, Intel oneAPI, Apple Metal pour Mac, et le calcul basé sur Vulkan. ROCm convient aux développeurs et aux organisations qui utilisent du hardware AMD pour l’IA.