【问题标题】:Installing tensorflow 1.8.0 gpu安装 tensorflow 1.8.0 gpu
【发布时间】:2022-04-11 18:00:45
【问题描述】:

我正在按照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

同样没有错误报告。

我尝试了 hello 这个词(或者到目前为止只有第一行:

>>> 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 设计的。

有什么建议吗?

【问题讨论】:

  • 我还在使用 TF 1.7,但是我在 TF 1.7 中遇到了类似的问题,它只支持 cuda 9.0,而 cuda 9.1 失败了。我怀疑这在 1.8 中可能没有改变,但我不确定。
  • 你是如何让它与 cuda 9.0 一起工作的,你同时安装了 9.0 和 9.1,还是只保留了 9.0?
  • 更新环境变量LD_LIBRARY_PATH,这就是tensorflow用来查找合适库的方法。但就我而言,我只是卸载了 9.1 并安装了 9.0,但我确信设置 LD_LIBRARY_PATH 也是一个可用的选项。
  • LD_LIBRARY_PATH 不适用于 Windows,但 PATH :)。反正刚刚找到一篇博文解决了。

标签: tensorflow


【解决方案1】:

在那里找到答案:https://www.codingforentrepreneurs.com/blog/install-tensorflow-gpu-windows-cuda-cudnn/

唯一不同的是我没有安装 CUDA 9.0 的驱动程序(仅 CUDA 部分)。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2018-12-20
    • 2019-03-14
    • 1970-01-01
    • 1970-01-01
    • 2020-12-20
    • 2018-07-12
    相关资源
    最近更新 更多