0
votes

Tensorflow 1.15 ne peut pas détecter le GPU avec CUDA10.1

J'ai installé les deux tensorflow 2.2.0 et tensorflow 1.15.0 (par PIP Installation tensorflow-gpu == 1.15.0 ) . Le tensorflow 2 est installé dans l'environnement de base de anaconda 3 , tandis que le tensorflow 1 est installé dans un environnement séparé.

Le TENSORFLOW 2.2.0 peut reconnaître le GPU en fonction d'un test simple: xxx

mais le flux tensorful 1.15.0 ne peut pas détecter le GPU.

Pour votre information, mon environnement système est Python + Cuda 10.1 + vs 2015.


0 commentaires

3 Réponses :


0
votes

Avez-vous essayé d'installer Anaconda? Il télécharge toutes les exigences et facilite la tâche en quelques clics.


1 commentaires

Oui, j'utilise Anaconda 3 et Tensorflows sont installés sur le dessus.



0
votes

Cela pourrait avoir à voir avec la compatibilité de la version de TF, Cuda et Cudnn. Cet article a-t-il discuté de manière approfondie.


3 commentaires

Merci! Mais il ne liste pas d'informations pour Tensorflow> = 1.15.0


Hmmm, à partir de ce forum, on dirait que vous avez besoin de Cuda10.0 Github.com/tensorflow/tensorflow/issues / 35926


Probablement. Je vais essayer cela, ainsi que la dernière version Tensorflow 1.15.3



1
votes

Les versions TENSOSFLOW 1.15.0 à 1.15.3 (la dernière version) sont toutes compilées contre CUDA 10.0. Rétrogradation du Cuda 10.1 à Cuda 10.0 a résolu le problème.

Soyez également au courant de la version Python. Il est recommandé d'installer le fichier tensorflow .whl (comme indiqué sur https://nero-mirror.stanford.edu/pypi/simple/tensorflow-gpu/ ) pour la version Python spécifique. Comme pour l'installation, voir Comment puis-je installer un Paquet Python avec un fichier .whl?


0 commentaires