CuDNN v6.0 fonctionnerait-il avec TensorFlow actuellement?
Avons-nous spécifiquement besoin de cuDNN v5.1 (comme suggéré) pour TensorFlow, ou la dernière version (v6.0) fonctionnerait-elle également? Existe-t-il une compatibilité ascendante dans les versions cuDNN?
Non, cuDNN 6.0 n'est pas pris en charge par la dernière version 1.2. Mais il y a un espoir: le fonctionnaire notes de publication dit ce qui suit:
TensorFlow 1.2 est peut-être la dernière fois que nous construisons avec CUDNN 5.1. À partir de TensorFlow 1.3, nous allons essayer de construire tous nos binaires pré-construits avec cuDNN 6.0. Bien que nous essayions de garder notre code source compatible avec cuDNN 5.1, ce sera au mieux.
J'espère donc que la prochaine version 1.3 utilisera CUDNN 6.0. Surtout maintenant, quand 7.0 est juste autour du coin .
Aujourd'hui (21 août 2017), j'ai installé la dernière version de Tensorflow v1.3 et je peux confirmer qu'elleREQUIERTcuDNN v6.0 et NE FONCTIONNERA PAS avec v5.1 . Il demandera en fait la bibliothèque libcudnn.so.6 et non la libcudnn.so.5
P.s. Si vous voulez que cela fonctionne avec cuDNN 5.1. vous pouvez installer une version précédente, par exemple v1.2 qui est à ce lien:
https://www.tensorflow.org/versions/r0.12/get_started/os_setup#virtualenv_installation
Ce serait une réponse mise à jour de la réponse de Salvador Dali .
J'ai mis à niveau la version tensorflow vers 1.2.1, puis cudnn 6.0 semble fonctionner sans problème. J'ai utilisé pip
pip install tensorflow
pip install --upgrade tensorflow
Les détails de l'installation sont ici .