Installation des bibliothèques ROCm sur GNU/Linux pour GPU AMD

Préparer votre GPU AMD pour les charges de travail IA nécessite la suite ROCm (Radeon Open Compute). Bien que le processus d'installation officiel soit simple, il y a un détail crucial qui peut vous éviter des instabilités système.

Le processus d'installation

Pour installer les bibliothèques et outils ROCm nécessaires, utilisez le script amdgpu-install fourni par AMD. Vous pouvez télécharger et installer la dernière version du paquet ROCm depuis le site officiel d'AMD : Pilotes GNU/Linux AMD.

L'astuce cruciale : évitez DKMS

Lors de l'exécution de la commande d'installation, il est fortement recommandé de ne pas installer les pilotes noyau DKMS. La version DKMS du pilote est souvent instable et peut entraîner des paniques noyau ou des problèmes de démarrage sur de nombreuses distributions GNU/Linux.

Utilisez plutôt le drapeau --no-dkms pour vous reposer sur les pilotes noyau intégrés.

La commande

Exécutez la commande suivante dans votre terminal :

amdgpu-install -y --usecase=rocm --no-dkms

Pourquoi utiliser --no-dkms ?

Le drapeau --no-dkms garantit que l'installateur ne tente pas de compiler et d'installer un module noyau séparé. Pour la plupart des utilisateurs, les pilotes déjà inclus dans le noyau GNU/Linux sont plus stables et suffisants pour le bon fonctionnement de ROCm.

Vérification

Après l'installation et un redémarrage du système, vous pouvez vérifier que votre GPU est reconnu par ROCm en exécutant :

rocminfo

Si la sortie liste correctement votre GPU, vous êtes prêt à commencer à exécuter des LLMs et d'autres modèles d'IA sur votre matériel AMD.