Installez la dernière version de TensorFlow prenant en charge CUDA, cudNN et GPU sur un PC Windows 11

introduire

Si vous êtes comme moi, vous savez que configurer TensorFlow sur un nouvel appareil peut être pénible. Dans ce didacticiel, je couvrirai l'installation de CUDA, cuDNN et TensorFlow sur Windows 11. Le but de cet article est de rendre ce processus aussi indolore que possible. Cet article suppose que vous disposez déjà d’un GPU compatible CUDA installé sur votre PC.

étapes d'installation

Il est important de suivre chaque étape dans l'ordre. Si vous effectuez les étapes dans le désordre, vous devrez recommencer le processus, ce que personne ne veut faire. Alors, sans plus attendre, allons droit au but.

Logiciels que nous devons installer :

Microsoft Visual Studio (2019 Community Edition)
NVIDIA CUDA Toolkit
NVIDIA cuDNN
Python (j'utilise personnellement anaconda)
TensorFlow

Étape 1 : Supprimez tous les pilotes NVIDIA actuels

Accédez à la barre de recherche sous Windows et recherchez « Ajouter ou supprimer des programmes ». Vous devriez maintenant voir une liste des applications installées. Faites défiler vers le bas jusqu'à ce que vous voyiez quelque chose qui dit NVIDIA. Cela devrait ressembler à l’image ci-dessous.

insérez la description de l'image ici
Maintenant, vérifiez et supprimez tout ce qui porte la mention NVIDIA. Ne vous inquiétez pas de les supprimer, nous les réinstallerons ultérieurement.

Étape 2 : Installez Visual Studio (pas le code VS)

L'installation de CUDA Toolkit nécessite Visual Studio. Si vous essayez de ne pas sécuriser

Je suppose que tu aimes

Origine blog.csdn.net/iCloudEnd/article/details/132423014
conseillé
Classement