Stable diffusion avec un GPU AMD sous Linux

Rédigé par Nicolas Sulek Aucun commentaire
Classé dans : Logiciel Mots clés : , amd, stable diffusion, rocm, python
Stable Diffusion, modèle d'apprentissage profond de génération d'images à partir de description en langage naturel, ne sait gérer par défaut que les GPU Nvidia grâce à CUDA et les CPU.

Pour qu'il puisse utiliser un GPU AMD sous Linux, il faut :
  • installer ROCM via la pile logicielle AMD
  • configurer pytorch pour l'utiliser.

Lire la suite