问题描述
尝试在我的机器上安装gpu版本的Tensorflow，但是在尝试'import tensorflow'时遇到此错误
运行Windows 10，Python 3.5.2，蟒蛇4.2.0 64位。我使用'pip install tensorflow-gpu'安装了tensorflow-gpu。
我尝试在，即运行Microsoft Visual C ++ 2015 Redistributable Update 3并确认我在System32中有以下DLL / SysWOW64 /我的Python文件夹：
值得注意的是，Tensorflow的cpu版本导入和运行正常，但是当我尝试使用gpu版本时，我得到以上错误。我认为这是因为我配置CUDA工具包/ cudNN的错误，但是我认为我在说明书中列出了正确的一切。我确定我在CUDA工具包的PATH环境变量中有一个条目（通向\bin和\libnvvp），并为cudNN添加一个条目（导致\cuda）。
您需要帮助的任何想法或其他信息？
尝试重命名 cudnn64_6.dll to cudnn64_5.dll in C：\Program Files\NVIDIA GPU Computing Toolkit\ CUDA\v8.0\bin 。
Trying to install the gpu version of Tensorflow on my machine, but I'm getting this error when trying 'import tensorflow'
Running Windows 10, Python 3.5.2, Anaconda 4.2.0 64-bit. Installed tensorflow-gpu using 'pip install tensorflow-gpu'.
I've attempted the fixes in https://github.com/tensorflow/tensorflow/issues/5949, namely running Microsoft Visual C++ 2015 Redistributable Update 3 and confirming I had the following DLLs in System32/SysWOW64/my Python folder:
It's worth noting that the cpu version of Tensorflow imports and runs just fine, but when I attempt to use the gpu version I get the above error. I presume this is due to a mistake on my part in configuring the CUDA toolkit/cudNN, but I think I did everything listed in the instructions correctly. I made sure I had an entry in the PATH environment variable for the CUDA toolkit (leading to \bin and \libnvvp) and added one in for cudNN (leading to \cuda).
Any thoughts or other information you'd need to help?
Try to rename cudnn64_6.dll to cudnn64_5.dll in C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin.
这篇关于无法在Windows 10上导入GPU的Tensorflow的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！