Python : tensorflow-gpu导入错误

Question

这是一个从关于链接的下列问题中延伸出来的问题：

python :无法导入tensorflow-gpu

我正试图在我的Windows 10机器上实现tensorflow-gpu。我遵循了这个链接让我做的事情。我有..。

1. 成功安装CUDA 8.0工具包
2. 成功安装cuDNN v5.1
3. 添加环境变量: CUDA_Home和路径变量
4. 安装了Anaconda3 (我以前使用Anaconda3)
5. 成功地执行了以下命令conda创建-n tensorflow- python=3.5.2 >activate tensorflow-gpu >pip install tensorflow-gpu >activate tensorflow-gpu ( tensorflow-gpu ) >python >>>

但是，如果我想进口tensorflow，

>>>import tensorflow as tf

我收到以下错误消息

Traceback (most recent call last):
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow-gpu\lib\site-
packages\tensorflow\python\platform\self_check.py", line 87, in preload_check
    ctypes.WinDLL(build_info.cudnn_dll_name)
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow-gpu\lib\ctypes\__init__.py", line 347, in __init__
self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] This 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\Froilan\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
from tensorflow.python import *
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 30, in <module>
self_check.preload_check()
  File "C:\Users\Froilan\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\platform\self_check.py", line 97, in preload_check
% (build_info.cudnn_dll_name, build_info.cudnn_version_number))
ImportError: Could not find 'cudnn64_6.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 6 from this URL: https://developer.nvidia.com/cudnn

我不知道是什么导致了这个问题。我正在使用AMD ryzen 1700 CPU & GTX 1050 Ti GPU。也许是其中之一造成了问题？

更新:带有CPU的tensorflow工作得很好。GPU是造成问题的原因

Answer 1

我相信使用命令

pip install tensorflow-gpu

安装TensorFlow的最新版本，目前的版本大约为1.6。根据我的经验，最新版本似乎只适用于CUDA9.0，这反过来需要cuDNN 7.0.4或7.0.5。您可以安装使用CUDA 8.0的特定版本的TensorFlow，例如，使用以下命令：

pip install tensorflow-gpu==1.4

或者，您可以从NVIDIA下载较早版本的CUDA和cuDNN。要做到这一点，您必须首先创建一个帐户(不要担心，它是免费的，非常快)。

旧版本的数据自动化系统在这里：https://developer.nvidia.com/cuda-toolkit-archive

这里cuDNN的旧版本：https://developer.nvidia.com/rdp/cudnn-download

TensorFlow网站可以进行更新，因为它仍然告诉您安装CUDA8.0，但我认为这只适用于TensorFlow版本1.4。

Answer 2

• 使用此命令conda install -c anaconda tensorflow-gpu
• 它为您安装cuda toolkit和cudnn，并执行所有配置。
• 如果要解决上述错误，请在cudnn变量中添加PATH路径。
• 您可以使用export PATH或转到.bashrc并在那里设置路径变量。

Answer 3

您的问题是tensorflow正在寻找cudnn v6。tensorflow 1.4与CUDA 8.0和CUDNN 6一起工作。默认情况下，pip将安装tensorflow 1.5/1.6，这在Windows上需要CUDA 9.0和CUDNN 6.0。检查需求这里。根据我的经验，错误的子版本(即6.1)也会导致问题。