python - 无法在 Windows 10 上导入 GPU 的 Tensorflow

Question

尝试在我的计算机上安装 GPU 版本的 Tensorflow，但在尝试“导入tensorflow”时遇到此错误

<小时/>

ImportError Traceback (most recent call last) C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflo w_internal.py in swig_import_helper() 17 try: ---> 18 return importlib.import_module(mname) 19 except ImportError:

C:\Users\snh5_000\Anaconda3\lib\importlib__init__.py in import_module(name, pac kage) 125 level += 1 --> 126 return _bootstrap._gcd_import(name[level:], package, level) 127

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in _gcd_import(name, pac kage, level)

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in _find_and_load(name, import_)

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in _find_and_load_unlock ed(name, import_)

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in _load_unlocked(spec)

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in module_from_spec(spec )

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap_external.py in create_modul e(self, spec)

C:\Users\snh5_000\Anaconda3\lib\importlib_bootstrap.py in _call_with_frames_rem oved(f, *args, **kwds)

ImportError: DLL load failed: The specified module could not be found.

During handling of the above exception, another exception occurred:

ImportError Traceback (most recent call last) C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflo w.py in () 40 sys.setdlopenflags(_default_dlopen_flags | ctypes.RTLD_GLOBAL) ---> 41 from tensorflow.python.pywrap_tensorflow_internal import * 42 from tensorflow.python.pywrap_tensorflow_internal import version

C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflo w_internal.py in () 20 return importlib.import_module('_pywrap_tensorflow_internal' ) ---> 21 _pywrap_tensorflow_internal = swig_import_helper() 22 del swig_import_helper

C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflo w_internal.py in swig_import_helper() 19 except ImportError: ---> 20 return importlib.import_module('_pywrap_tensorflow_internal' ) 21 _pywrap_tensorflow_internal = swig_import_helper()

C:\Users\snh5_000\Anaconda3\lib\importlib__init__.py in import_module(name, pac kage) 125 level += 1 --> 126 return _bootstrap._gcd_import(name[level:], package, level) 127

ImportError: No module named '_pywrap_tensorflow_internal'

During handling of the above exception, another exception occurred:

ImportError Traceback (most recent call last) in () ----> 1 import tensorflow

C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow__init__.py in () 22 23 # pylint: disable=wildcard-import ---> 24 from tensorflow.python import * 25 # pylint: enable=wildcard-import 26

C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python__init__.py in < module>() 49 import numpy as np 50 ---> 51 from tensorflow.python import pywrap_tensorflow 52 53 # Protocol buffers

C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflo w.py in () 50 for some common reasons and solutions. Include the entire stack trace 51 above this error message when asking for help.""" % traceback.format_exc () ---> 52 raise ImportError(msg) 53 54 # pylint: enable=wildcard-import,g-import-not-at-top,unused-import,line- too-long

ImportError: Traceback (most recent call last): File "C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_t ensorflow_internal.py", line 18, in swig_import_helper return importlib.import_module(mname) File "C:\Users\snh5_000\Anaconda3\lib\importlib__init__.py", line 126, in imp ort_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 986, in _gcd_import File "", line 969, in _find_and_load File "", line 958, in _find_and_load_unlocked File "", line 666, in _load_unlocked
File "", line 577, in module_from_spec
File "", line 906, in create_module File "", line 222, in _call_with_frames_removed ImportError: DLL load failed: The specified module could not be found.

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_t ensorflow.py", line 41, in from tensorflow.python.pywrap_tensorflow_internal import * File "C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_t ensorflow_internal.py", line 21, in _pywrap_tensorflow_internal = swig_import_helper() File "C:\Users\snh5_000\Anaconda3\lib\site-packages\tensorflow\python\pywrap_t ensorflow_internal.py", line 20, in swig_import_helper return importlib.import_module('_pywrap_tensorflow_internal') File "C:\Users\snh5_000\Anaconda3\lib\importlib__init__.py", line 126, in imp ort_module return _bootstrap._gcd_import(name[level:], package, level) ImportError: No module named '_pywrap_tensorflow_internal'

Failed to load the native TensorFlow runtime.

See https://www.tensorflow.org/install/install_sources#common_installation_probl ems

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

运行 Windows 10、Python 3.5.2、Anaconda 4.2.0 64 位。使用“pip install tensorflow-gpu”安装了tensorflow-gpu。

我已尝试修复 https://github.com/tensorflow/tensorflow/issues/5949 ，即运行 Microsoft Visual C++ 2015 Redistributable Update 3 并确认 System32/SysWOW64/my Python 文件夹中有以下 DLL:

KERNEL32.dll

WSOCK32.dll

WS2_32.dll

SHLWAPI.dll

python35.dll

MSVCP140.dll

VCRUNTIME140.dll

api-ms-win-crt-runtime-l1-1-0.dll

api-ms-win-crt-heap-l1-1-0.dll

api-ms-win-crt-utility-l1-1-0.dll

api-ms-win-crt-stdio-l1-1-0.dll

api-ms-win-crt-string-l1-1-0.dll

api-ms-win-crt-math-l1-1-0.dll

api-ms-win-crt-convert-l1-1-0.dll

api-ms-win-crt-environment-l1-1-0.dll

api-ms-win-crt-filesystem-l1-1-0.dll

api-ms-win-crt-time-l1-1-0.dll

值得注意的是，CPU 版本的 Tensorflow 导入并运行得很好，但是当我尝试使用 GPU 版本时，出现上述错误。我认为这是由于我在配置 CUDA 工具包/cudNN 时犯了一个错误，但我认为我正确执行了说明中列出的所有操作。我确保我在 CUDA 工具包的 PATH 环境变量中有一个条目(通向\bin 和\libnvvp)，并为 cudNN 添加了一个条目(通向\cuda)。

您有什么想法或其他信息需要帮助吗？

Answer 1

尝试将C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v8.0\bin中的cudnn64_6.dll重命名为cudnn64_5.dll.