我在Windows 10中运行Tensorflow-gpu遇到问题

时间:2020-01-18 01:17:27

标签: python-3.x tensorflow

我有一个GTX 1650和一个Windows10。我为CUDA 10.1下载了CUDA 10.1和cuDNN v7.6.4(2019年9月27日)。遵循了教程之后,我按照教程上的指示进行了操作。

完成后,我运行了一个脚本进行检查:

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

我收到以下错误:

跟踪(最近一次通话最近):文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow.py”, 第58行,在 从tensorflow.python.pywrap_tensorflow_internal导入*文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow_internal.py”, 第28行,在 _pywrap_tensorflow_internal = swig_import_helper()文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow_internal.py”, 第24行,在swig_import_helper中 _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路径名,描述)文件 “ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ imp.py”,第243行,在 load_module 返回load_dynamic(名称,文件名,文件)文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ imp.py”,第343行,在 load_dynamic return _load(spec)ImportError:DLL加载失败:找不到指定的模块。

在处理上述异常期间,发生了另一个异常:

回溯(最近一次通话最后一次):文件“ pypy.py”,位于第1行 从tensorflow.python.client导入device_lib文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow__init __。py”, 第101行,在 从tensorflow_core导入*文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core__init __。py”, 第40行,在 从tensorflow.python.tools导入module_util作为_module_util文件 “ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow__init __。py”, 第50行,位于 getattr module = self._load()文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow__init __。py”, _load中的第44行 module = _importlib.import_module(自己。名称)文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ importlib__init __。py”, import_module中的第126行 返回_bootstrap._gcd_import(名称[级别:],包,级别)文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python__init __。py”, 第49行,在 从tensorflow.python导入pywrap_tensorflow文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow.py”, 74行,在 引发ImportError(msg)ImportError:追溯(最近一次调用):文件 “ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow.py”, 第58行,在 从tensorflow.python.pywrap_tensorflow_internal导入*文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow_internal.py”, 第28行,在 _pywrap_tensorflow_internal = swig_import_helper()文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ site-packages \ tensorflow_core \ python \ pywrap_tensorflow_internal.py”, 第24行,在swig_import_helper中 _mod = imp.load_module('_ pywrap_tensorflow_internal',fp,路径名,描述)文件 “ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ imp.py”,第243行,在 load_module 返回load_dynamic(名称,文件名,文件)文件“ C:\ Users \ prono \ Anaconda3 \ envs \ tensorflow \ lib \ imp.py”,第343行,在 load_dynamic return _load(spec)ImportError:DLL加载失败:找不到指定的模块。

无法加载本机TensorFlow运行时。

请参见https://www.tensorflow.org/install/errors

出于某些常见原因和解决方案。寻求帮助时,请在此错误消息上方包含整个堆栈跟踪。

P.S。这是我第一次发布问题,如果我不清楚该帖子,请原谅我。我只是不知道还要添加什么

1 个答案:

答案 0 :(得分:-1)

对于Windows 10:

https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

对于ubuntu 18.04,请查看我的cuda安装手册:

https://gist.github.com/sailfish009/2f542992a67da726f41dcc9692da3f56

-

https://github.com/tensorflow/tensorflow/issues/22794

尝试在Windows 10上使用tensorflow-gpu时遇到相同的问题。 由于我无法在cuDNN10上使用它,因此我选择了9 使用...

conda create --name tf-gpu
conda install -c aaronzs tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
conda install keras-gpu

您还可以在以下位置查看完整帖子 https://www.pugetsystems.com/labs/hpc/The-Best-Way-to-Install-TensorFlow-with-GPU-Support-on-Windows-10-Without-Installing-CUDA-1187/

唐纳德·金霍恩博士的信