web-dev-qa-db-fra.com

Erreur d'importation Tensorflow

J'essaye d'installer tensorflow qui supporte le GPU.

J'ai essayé l'information dans le lien suivant

https://www.tensorflow.org/install/install_windows

  1. CUDA® Toolkit 8.0
  2. cUDNN v6.0
  3. Carte GPU avec capacité de calcul CUDA 3.0 - GeForce 940MX

Ensuite, utilisez pip3 install --upgrade tensorflow-gpu pour installer tensorflow.

Mais je reçois l'erreur suivante lorsque j'essaie d'importer tensorflow.

    Traceback (most recent call last):
      File "C:\Research\Python_installation\lib\site-packages\tensorflow\python\platform\self_check.py", line 75, in preload_check
        ctypes.WinDLL(build_info.cudart_dll_name)
      File "C:\Research\Python_installation\lib\ctypes\__init__.py", line 347, in __init__
        self._handle = _dlopen(self._name, mode)
    OSError: [WinError 126] The specified module could not be found

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):
      File "<pyshell#6>", line 1, in <module>
        import tensorflow as tf
      File "C:\Research\Python_installation\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
        from tensorflow.python import *
      File "C:\Research\Python_installation\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
        from tensorflow.python import pywrap_tensorflow
      File "C:\Research\Python_installation\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
        self_check.preload_check()
      File "C:\Research\Python_installation\lib\site-packages\tensorflow\python\platform\self_check.py", line 82, in preload_check
        % (build_info.cudart_dll_name, build_info.cuda_version_number))
    ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit

Quelqu'un peut-il s'il vous plaît me conseiller sur la façon de l'utiliser.

pythontensorflowcuda
9
user3789200

J'ai eu un problème similaire et je devais faire très attention à la version de CUDA et à la version de CuDNN. J'ai frappé l'erreur exacte que vous avez frappée, et je l'ai corrigé en parcourant ce que j'ai documenté ici: http://www.laurencemoroney.com/installing-tensorflow-with-gpu-on-windows-10/

Essaie! :)

(L'échec le plus courant que j'ai constaté est que vous téléchargez le dernier CUDA, et non le correspondant CUDA. À l'heure actuelle, CUDA est à 9.1, mais TF requiert 9.0 - votre erreur dit cudart90.dll ) - pour trouver les pilotes 9.0, téléchargez-les et installez-les. Ensuite, lancez TensorFlow. Cela pourrait alors échouer sur les pilotes CuDNN, ce qui est bien, car vous savez que CUDA a raison. Téléchargez ensuite les bons pilotes CuDNN (correspondant au numéro de version de l’erreur) et réessayez.)

9
Laurence Moroney

S'il vous plaît télécharger les pilotes appropriés car la dernière version de cuda v8.0 toolkit a besoin de tensorflow 1.4.x.

Comme par défaut, pip3 install --upgrade tensorflow-gpu installe la dernière version de tensorflow qui nécessite cuda v9.0 (last) et non cuda v8.0 toolkit

Vous devez installer la version 1.4.1 pour CUDA-8 en tant que

pip install tensorflow-gpu==1.4.1

La dernière version (version 1.5) est destinée à CUDA-9 Vous devrez installer la version 1.5 pour CUDA-9 en tant que

pip install tensorflow-gpu==1.5

Téléchargez cuddn pour votre version appropriée de cuda toolkit à partir de ici Et suivez this .

Téléchargez le cuda toolkit approprié depuis ici

0
susan097