web-dev-qa-db-fra.com

Comment installer TensorFlow-gpu avec cuda8.0?

J'ai essayé de l'installer conformément aux instructions du site Web officiel, ce qui entraîne une erreur ImportError lors de l'importation du tensorflow:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

Je lance le code cat /usr/local/cuda/version.txt, qui indique que ma version cuda est 8.0.61.

Il semble que tensorflow recherche Cuda 9.0. Je ne peux pas mettre à jour le cuda car je travaille sur un serveur gpu partagé et je n'ai pas l'autorité racine.

Est-il possible de faire fonctionner tensorflow avec cuda 8.0? Ou de toute autre manière disponible?

Merci!!

21
inkzk

Je faisais face au même problème, jusqu'à ce que je trouve

https://www.tensorflow.org/install/install_sources#tested_source_configurations

vérifiez votre version de cuda installée et votre version de cudnn, puis déterminez quelle version de tensorflow-gpu est compatible avec ceux qui utilisent le lien mentionné ci-dessus.

J'avais installé cuda 8 et cudnn v5.1, donc en vérifiant le lien ci-dessus, tensorflow-gpu 1.2.0 était compatible et après avoir installé

pip install tensorflow-gpu==1.2.0

Cela a fonctionné pour moi.

12
Azam Rafique

Vous devrez installer la version 1.4.1 pour CUDA-8 comme

pip install tensorflow-gpu==1.4.1

La dernière version 1.5) est pour CUDA-9

39
layog