无法在 python 3.6 中导入 TensorFlow

Question

我无法导入 Tensorflow。我的 gpu nvidia 940mx 和我正在使用 python 3.6。

我安装的软件包是： 绝对py（0.2.0） 阿斯特（0.6.2） 漂白剂 (1.5.0) 循环仪 (0.10.0) 加气（0.2.0） grpcio (1.11.0) html5lib (0.9999999) 奇异果求解器 (1.0.1) 降价（2.6.11） matplotlib (2.2.2) 麻木（1.14.2） opencv-python (3.4.0.12) 点子（9.0.3） protobuf (3.5.2.post1) pyparsing (2.2.0) python-dateutil (2.7.2) 皮茨（2018.4） 安装工具 (39.0.1) 六（1.11.0） 张量板（1.7.0） 张量流GPU（1.7.0） 术语颜色（1.1.0） Werkzeug (0.14.1) 轮子 (0.31.0)

错误是：

import tensorflow as tf
    Traceback (most recent call last):
      File "C:\Program Files\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 75, in preload_check
        ctypes.WinDLL(build_info.cudart_dll_name)
      File "C:\Program Files\Python36\lib\ctypes\__init__.py", line 348, 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 "<stdin>", line 1, in <module>
  File "C:\Program Files\Python36\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
    from tensorflow.python import *  # pylint: disable=redefined-builtin
  File "C:\Program Files\Python36\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:\Program Files\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
    self_check.preload_check()
  File "C:\Program Files\Python36\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`enter code here`

Answer 1

您安装的 Tensorflow 版本需要 CUDA 9.0。您应该检查是否安装了它（而不是其他版本）。

如果你想使用例如CUDA 9.1，你需要自己编译 Tensorflow（教程here）。