Tensorflow导入错误

Question

我正在尝试安装支持GPU的tensorflow.

我尝试了以下链接中的信息

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

1. CUDA®Toolkit8.0
2. cuDNN v6.0
3. 具有CUDA Compute Capability 3.0的GPU卡 - GeForce 940MX

然后用来pip3 install --upgrade tensorflow-gpu安装tensorflow.

但是在尝试导入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

有人可以请教我如何使用它.

Answer 1

我有类似的问题,并且必须非常小心CUDA的版本和CuDNN的版本.我点击了你正在发生的确切错误,并通过我在此处记录的内容修复它:http://www.laurencemoroney.com/installing-tensorflow-with-gpu-on-windows-10/

试试看!:)

(我发现最常见的故障是你下载了最新的 CUDA,而不是匹配的 CUDA.现在CUDA是9.1,但TF需要9.0 - 你的错误说cudart90.dll) - 所以找到9.0驱动程序,下载并安装它们.然后运行TensorFlow.它可能会在CuDNN驱动程序上失败,这很好,因为你知道CUDA是对的.然后下载正确的CuDNN驱动程序(匹配错误中的版本号),然后重试.)