Question

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

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

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

CUDA®Toolkit8.0
cuDNN v6.0
带有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驱动程序（匹配错误中的版本号）并再试一次。）

Answer 2

请下载正确的驱动程序，因为最新版本的cuda v8.0工具箱需要tensorflow 1.4.x需求。

因此，默认情况下pip3 install --upgrade tensorflow-gpu安装需要CUDA V9.0（最新版本）而不是CUDA V8.0工具包的tensorflow的最新版本。

您需要将CUDA-8的1.4.1版本安装为

pip install tensorflow-gpu==1.4.1

最新版本（1.5版）适用于CUDA-9 您需要将CUDA-9的版本1.5安装为

pip install tensorflow-gpu==1.5

从here下载适合您的cuda版本工具包的cuddn 并遵循this。

从here下载适当的cuda工具包

Tensorflow导入错误

2 个答案: