Python : tensorflow-gpu import errors

这是从链接上的以下问题扩展而来的问题:

python : cannot import tensorflow-gpu

我正在尝试在我的 Windows 10 机器上实现 tensorflow-gpu。我按照 link 告诉我的去做。我有..

成功安装 CUDA 8.0 Toolkit

成功安装cuDNN v5.1

添加了环境变量:CUDA_Home 和路径变量

安装了 Anaconda3(我以前使用的是 Anaconda3)

成功执行了以下命令

>conda create -n tensorflow-gpu 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 是导致问题的原因

最佳答案

我相信使用命令

pip install tensorflow-gpu

安装最新版本的 TensorFlow，目前类似于 1.6。根据我的经验，最新版本似乎只适用于 CUDA 9.0，而后者又需要 cuDNN 7.0.4 或 7.0.5。
您可以安装可与 CUDA 8.0 一起使用的特定版本的 TensorFlow，例如，使用以下命令:

pip install tensorflow-gpu==1.4

或者，您可以从 NVIDIA 下载旧版本的 CUDA 和 cuDNN。为此，您必须先创建一个帐户(别担心，它是免费且非常快速的)。

旧版本的 CUDA 在这里:https://developer.nvidia.com/cuda-toolkit-archive

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

TensorFlow 网站可以进行更新，因为它仍然会告诉您安装 CUDA 8.0，但我认为这仅适用于 TensorFlow 1.4 版。

关于Python : tensorflow-gpu import errors，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/48075842/

Tensorflow

Python : tensorflow-gpu import errors