tensorflow-gpu ImportError: DLL load failed: 找不到指定的模块。解决办法

最近好不容易弄了块1070的显卡,准备大展身手,做gpu训练。

调用之后会发现出现imprortError ImportError: DLL load failed: 找不到指定的模块。

具体错误是Traceback (most recent call last):
  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python36\lib\imp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python36\lib\imp.py", line 342, in load_dynamic
    return _load(spec)

网上看了好多博客,都没有解决。其实呢,这是你直接pip tensorflow-gpu 所导致的问题。

pip得到的tensorflow 支持的是CUDA9版本,但是最新的已经是10了,怎么办呢?

不用重新把10换成 9,只需要3步

第一步 打开cmd 输入输入nvcc -V,看到下图内容就成功了

第二步 Pip install tensorflow 注意这里是cpu版本

第三部 下载网上大神们准备好的编译支持cuda10的tensoflow-gpu 版本 https://github.com/fo40225/tensorflow-windows-wheel/blob/master/1.12.0/py36/GPU/cuda100cudnn73sse2/tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl

然后下载好 pip install 存放路径\tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl 

然后就成功啦!

猜你喜欢

转载自blog.csdn.net/PythonstartL/article/details/87881684