在Windows上安装Tensorflow的问题

时间:2017-09-27 20:40:34

标签: python tensorflow

在安装了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_HOMECUDA_PATHCUDA_PATH_V8_0。此外,cuDNN已使用nVidia的指令安装,路径变量已设置为CUDA\v8.0\binCUDA\v8.0\libnvvp

1 个答案:

答案 0 :(得分:1)

这可能是因为Tensorflow现在只支持cuDNN v6.0或v6.1,至少是Installation Guide for Windows中保留的内容。

我遇到了同样的问题,但是将Tensorflow从旧版本更新为更新版本的cuDNN必须更新。