2017-09-27 274 views
0

在安装了CUDA 8和CuDNN 7的Windows 10上,安装Tensorflow(GPU和CPU版本,尽管我将专注于GPU版本)存在麻烦。在Windows上安装Tensorflow的问题

当试图用python3.5使用pip install tensorflow-gpu在系统范围内安装它时,它报告没有匹配的软件包。

如果我用整个社会的支持蟒蛇分布(使用在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_PATH,并通过CUDA_PATH_V8_0各种教程的建议此外cuDNN一直使用NVI安装dia的指令和路径变量已被设置为CUDA\v8.0\binCUDA\v8.0\libnvvp

回答

1

这可能是因为Tensorflow现在只支持cuDNN v6.0或v6.1,至少在Installation Guide for Windows中维护的是什么。

我有同样的问题,但在更新Tensorflow从旧版本到更新的cuDNN必须更新之后。

+0

看来我已经正确安装了cuDNN v6.0。 (我有'cudnn64_7.dll',它的依赖放置,并链接到路径变量) –

+0

有说:“特别是,cuDNN版本必须完全匹配:如果找不到'cuDNN64_6.dll '要使用不同版本的cuDNN,你必须从源代码构建。“ –

+0

谢谢,你似乎是正确的,它是由于在cuDNN库的版本中引起混淆而引起的。 –