【发布时间】: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