首页 文章

安装tensorflow 1.8.0 gpu

提问于
浏览
0

我按照https://www.tensorflow.org/install/install_windows上的说明使用pip安装tensorflow gpu

pip3 install --upgrade tensorflow-gpu

这似乎工作,并表明已安装版本1.8.0 .

唯一的警告是关于pip不是最新的,因为我建议用以下命令更新pip:

python -m pip install --upgrade pip

同样没有错误报告 .

我试过你好的话(或者到目前为止只有第一行:

>>> import tensorflow as tf

我收到一条很长的错误信息:

Traceback (most recent call last):
  File "C:\Users\Me\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py", line 75, in preload_check
    ctypes.WinDLL(build_info.cudart_dll_name)
  File "C:\Users\Me\AppData\Local\Programs\Python\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:\Users\Me\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "C:\Users\Me\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:\Users\Me\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
    self_check.preload_check()
  File "C:\Users\Me\AppData\Local\Programs\Python\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

我安装了Cuda 9.1,因为其他软件包需要该版本 . 虽然说明页面显示Cuda 9.0,但说明页面上的链接链接到nvidia站点上的Cuda 9.1 .

所以问题是有一种方法可以安装cuda 9.1,或者我应该等到另一个版本可用 . 在浏览其他类似的问题时,似乎有一段时间它被计划为tensorflow 1.5 .

有什么建议吗?

1 回答

相关问题