Actuellement conda install tensorflow-gpu
installe tensorflow v2.3.0 et n'installe PAS les packages conda cudnn ou cudatoolkit. Les installer manuellement (par exemple avec conda install cudatoolkit=10.1
) ne semble pas résoudre le problème non plus.
Une solution consiste à installer une version antérieure de tensorflow, qui installe cudnn et cudatoolkit, puis à mettre à niveau avec pip
conda install tensorflow-gpu=2.1
pip install tensorflow-gpu==2.3.1
(2.4.0 utilise cuda 11.0 et cudnn 8.0, cependant cudnn 8.0 n'est pas dans anaconda au 16/12/2020)
Edit :veuillez également consulter la réponse de @ GZ0, qui renvoie à une discussion sur github avec une solution en une seule ligne
Le tensorflow
build automatiquement sélectionné par Anaconda sur Windows 10 lors de l'installation de tensorflow-gpu
2.3 semble défectueux. Veuillez trouver une solution de contournement ici (envisagez de voter pour la réponse GitHub si vous avez un compte GitHub).
Je n'ai pas non plus (encore) réussi à faire en sorte que TF 2.3.0 reconnaisse mon GPU Nvidia Quadro Pro 620.
Note :J'ai 2 autres 'environnements' sur ce PC (windows Pro) Tous installés via Anaconda :
- Python 3.7.8 TF 2.0.0... reconnaît (et utilise) le GPU Nvidia
- Python 3.6.9 TF 2.1.0... reconnaît (et utilise) le GPU Nvidia
- Python 3.8.6 TF 2.3.0... ne voit PAS le GPU
Ma machine a Cuda 11.1; cuDNN 8.0.5
Ma prochaine pensée est d'envisager de rétrograder Python de 3.8.6 à 3.7.8 dans la 3ème configuration où TF =2.3.0
Steve