Cannot import Tensorflow for GPU on Windows 10

后端 未结 9 1740
被撕碎了的回忆
被撕碎了的回忆 2020-11-27 08:16

Trying to install the gpu version of Tensorflow on my machine, but I\'m getting this error when trying \'import tensorflow\'


ImportError

相关标签:
9条回答
  • 2020-11-27 08:19

    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.

    0 讨论(0)
  • 2020-11-27 08:21

    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.

    0 讨论(0)
  • 2020-11-27 08:21

    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.

    0 讨论(0)
  • 2020-11-27 08:22

    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.

    0 讨论(0)
  • 2020-11-27 08:24

    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

    0 讨论(0)
  • 2020-11-27 08:26

    Try to rename cudnn64_6.dll to cudnn64_5.dll in C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin.

    0 讨论(0)
提交回复
热议问题