【发布时间】:2019-02-08 12:44:47
【问题描述】:
我想使用 tensorflow GPU 版本。我已经通过 pip install 命令安装了 tensorflow,它安装了 1.10 的最新版本。观看视频后,我为 CUDA 9.2 安装了 CUDA 工具包最新版本 9.2 和 cudnn 版本 7.2.1。导入时出现错误
Traceback(最近一次调用最后一次): 文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py”,第 75 行,在 preload_check ctypes.WinDLL(build_info.cudart_dll_name) init 中的文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\ctypes__init__.py”,第 348 行 self._handle = _dlopen(self._name, mode) OSError: [WinError 126] 找不到指定的模块
在处理上述异常的过程中,又发生了一个异常:
Traceback(最近一次调用最后一次): 文件“”,第 1 行,在 导入张量流 文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow__init__.py”,第 22 行,在 from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import 文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python__init__.py”,第 49 行,在 从 tensorflow.python 导入 pywrap_tensorflow 文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py”,第 30 行,在 self_check.preload_check() 文件“C:\Users\Lenovo\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\self_check.py”,第 82 行,在 preload_check % (build_info.cudart_dll_name, build_info.cuda_version_number)) ImportError:找不到“cudart64_90.dll”。 TensorFlow 要求将此 DLL 安装在以 %PATH% 环境变量命名的目录中。从以下 URL 下载并安装 CUDA 9.0:https://developer.nvidia.com/cuda-90-download-archive
无法理解为什么它要求使用旧版本 cudart64_90。我必须安装 CUDA 9.0 还是可以对现有安装进行一些更改。
【问题讨论】:
-
目前pypi预建的tensorflow只支持cuda9.0。 cudnn7.如果你想要 cuda9.2,也许你可以从源代码构建
-
你能告诉我从源代码构建是什么意思吗? @Bugkiller
-
@Ishaan 从源代码构建,他的意思是从 tensorflow 源代码自己编译/安装它。基于你不得不问的事实,我认为你还没有准备好迈出这样的一步。我认为如果你只是恢复到 CUDA 9.0,你会有更好的运气。
标签: python tensorflow