Question

我安装了CUDA因为我需要运行一些Deep Neural Network模型，但是Tensorflow仍然无法看到gpu

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

它只显示CPU。

如何让Tensorflow使用我的gpu？

错误日志我在安装了tensorflow-gpu导入keras时得到了 ImportError：Traceback（最近一次调用最后一次）： swig_import_helper中的文件“C：\ Users \ Monviso \ Anaconda3 \ lib \ site-packages \ tensorflow \ python \ pywrap_tensorflow_internal.py”，第18行 return importlib.import_module（mname）在import_module中输入文件“C：\ Users \ Monviso \ Anaconda3 \ lib \ importlib__init __。py”，第126行 return _bootstrap._gcd_import（name [level：]，package，level）文件“”，第978行，在_gcd_import中文件“”，第961行，在_find_and_load中文件“”，第950行，在_find_and_load_unlocked中文件“”，第648行，在_load_unlocked中在module_from_spec中的文件“”，第560行在create_module中的文件“”，第922行文件“”，第205行，在_call_with_frames_removed中 ImportError：DLL加载失败：找不到指定的模块。

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\Users\Monviso\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\Users\Monviso\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\Users\Monviso\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
    return importlib.import_module('_pywrap_tensorflow_internal')
  File "C:\Users\Monviso\Anaconda3\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
ModuleNotFoundError: No module named '_pywrap_tensorflow_internal'


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.

Answer 1

你的Cuda版本就是问题所在。只有tensorflow版本1.5支持CUDA 9和cuDNN 7.以前版本的Tensorflows仅支持CUDA 8和cuDNN 6.如果您使用anaconda安装，很可能您没有安装1.5版本。使用pip或pip3进行安装。

由于您使用的是Windows机器，请检查this链接以使用gpu支持安装tensorflow。

检查此link以查看哪个版本需要Cuda和Cudnn

我建议您卸载tensorflow的CPU和GPU版本并重新安装。

如果你想保持你的cuda版本为9，那么安装tensorflow版本1.5。如果使用anaconda进行安装，很可能会获得仅支持Cuda 8的版本1.4

Tensorflow无法使用GPU

1 个答案: