在安装了CUDA 8和CuDNN 7的Windows 10上,我在安装Tensorflow时遇到了麻烦(包括GPU和CPU版本,但我会专注于GPU版本)。
当尝试使用pip install tensorflow-gpu
使用python3.5在系统范围内安装它时,它报告没有匹配的包。
如果我使用社区支持的anaconda发行版安装它(使用documentation页面中描述的步骤,它会正确安装Tensorflow,但是当我将其导入程序时会显示以下错误:
>>> import tensorflow as tf
[...]
ImportError: DLL load failed: The specified module could not be found.
[...]
ImportError: No module named '_pywrap_tensorflow_internal'
[...]
Failed to load the native TensorFlow runtime.
paste-bin上提供了整个堆栈跟踪。
对于CUDA,我按照各种教程的建议设置了以下系统环境变量:CUDA_HOME
,CUDA_PATH
和CUDA_PATH_V8_0
。此外,cuDNN已使用nVidia的指令安装,路径变量已设置为CUDA\v8.0\bin
和CUDA\v8.0\libnvvp
。
答案 0 :(得分:1)
这可能是因为Tensorflow现在只支持cuDNN v6.0或v6.1,至少是Installation Guide for Windows中保留的内容。
我遇到了同样的问题,但是将Tensorflow从旧版本更新为更新版本的cuDNN必须更新。