tensorflow-gpu的ImportError

时间:2018-07-07 12:38:19

标签: python tensorflow

我在WIN10 cp上成功安装了Tensorflow,但是当我尝试导入Tensorflow时,出现导入错误,如下所示:

    import tensorflow
    Traceback (most recent call last):
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 87, in preload_check
        ctypes.WinDLL(build_info.cudnn_dll_name)
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\ctypes\__init__.py", line 348, in __init__
        self._handle = _dlopen(self._name, mode)
    OSError: [WinError 126] 找不到指定的模块。

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
        from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
        from tensorflow.python import pywrap_tensorflow
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
        self_check.preload_check()
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 97, in preload_check
        % (build_info.cudnn_dll_name, build_info.cudnn_version_number))
    ImportError: Could not find 'cudnn64_7.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 7 from this URL: https://developer.nvidia.com/cudnn
    >>> import tensorflow
    Traceback (most recent call last):
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
        from tensorflow.python.pywrap_tensorflow_internal import *
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in <module>
        _pywrap_tensorflow_internal = swig_import_helper()
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 17, in swig_import_helper
        return importlib.import_module(mname)
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\importlib\__init__.py", line 126, in import_module
        return _bootstrap._gcd_import(name[level:], package, level)
    ImportError: DLL load failed: 找不到指定的模块。

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
        from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
        from tensorflow.python import pywrap_tensorflow
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 74, in <module>
        raise ImportError(msg)
    ImportError: Traceback (most recent call last):
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
        from tensorflow.python.pywrap_tensorflow_internal import *
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in <module>
        _pywrap_tensorflow_internal = swig_import_helper()
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 17, in swig_import_helper
        return importlib.import_module(mname)
    File "C:\Users\25457\AppData\Local\Programs\Python\Python36\lib\importlib\__init__.py", line 126, in import_module
        return _bootstrap._gcd_import(name[level:], package, level)
    ImportError: DLL load failed: 找不到指定的模块。


    Failed to load the native TensorFlow runtime.

    See https://www.tensorflow.org/install/install_sources#common_installation_problems

    for some common reasons and solutions.  Include the entire stack trace
    above this error message when asking for help.

我已经安装了CUDA v9.2和cuDNN 7,也可以找到“ pywrap_tensorflow_internal.py”,当我安装了Tensorflow的CPU版本时,它运行的很好,但它不适用于tensorflow gpu。可能出什么问题了?

1 个答案:

答案 0 :(得分:1)

问题解决了。 我用7.0.4重新安装了CUDA V9.0和cuDNN 然后用重新安装tensorflow-gpu     pip卸载tensorflow-gpu     pip install tensorflow-gpu