Trying to install the gpu version of Tensorflow on my machine, but I\'m getting this error when trying \'import tensorflow\'
ImportError
I still got the _pywrap_tensorflow_internal
error after installing cuDNN 5.1 (yes, I had PATH set correctly). I had better luck installing tensorflow_gpu from this site. This requires cuDNN 6.0, not 5.1. Unfortunately they currently only have Tensorflow 1.1 for GPU systems.
Have you tried turning it off and on again? Seriously, though. If you have your command line terminal running during the CUDA and cuDNN installs, the environment variables aren't refreshed until you close and re-open the terminal. I know it's simple, but I struggled for an hour before a simple restart solved that same import error.
I tried all the solutions here and it wasn't working for me. Check this link for the requirements and make sure you have the compatible versions. I also added all the cuDNN paths to my environment variables and also don't forget to copy the contents of lib,bin and include to the same folders inside your CUDA installation.
This is what worked for me. I referred to the Tensorflow website to find the currently supported CudaNN version. When I solved my issue, it was CudaNN 6.x. The latest version, in my case 7 DLL is not loaded by tensorflow, and if you try to rename it, it causes memory errors.
I had the same problem running Windows 10, Anaconda 4.3.21 64bit, Python 3.5.3, and TensorFlow 1.2.0.
However, the issue was fixed by installing cuDNN v5.1 instead of v6.0 as proposed here https://stackoverflow.com/a/43566341/3830240
Try to rename cudnn64_6.dll
to cudnn64_5.dll
in C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
.