Tensorflow无法使用GPU

2024-04-26 23:32:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我安装了CUDA,因为我需要运行一些深层神经网络模型,但是Tensorflow仍然无法看到gpu

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

它只显示CPU。在

如何让Tensorflow使用我的gpu?在

导入安装了tensorflow gpu的keras时收到的错误日志 最近的恐怖事件(回溯:上次导入): 文件“C:\Users\Monviso\Anaconda3\lib\site packages\tensorflow\python\pywrap_tensorflow_内部.py,第18行,在swig_import_helper中 返回importlib.import_模块(名称) 文件“C:\Users\Monviso\Anaconda3\lib\importlib_uinit_uy.py”,第126行,在import_模块中 return _bootstrap.\u gcd_import(名称[级别:],包,级别) 文件“”,第978行,在“导入”中 文件“”,第961行,在“查找”和“加载”中 文件“”,第950行,在“查找”和“加载”中解锁 文件“”,第648行,在“加载”中解锁 文件“”,第560行,来自规范的模块 文件“”,第922行,在create\u模块中 文件“”,第205行,在“调用”中删除了“帧” ImportError:DLL加载失败:找不到指定的模块。在

^{pr2}$

Tags: 模块文件pyimport名称gpudevicelib
1条回答
网友
1楼 · 发布于 2024-04-26 23:32:29

你的Cuda版本就是问题所在。只有tensorflow 1.5版支持CUDA 9和cuDNN 7。以前版本的Tensorflows仅支持CUDA 8和cuDNN 6。如果您使用anaconda安装,很可能您没有安装版本1.5。使用pip或pip3安装。在

由于您使用的是windows机器,请检查this链接来安装支持gpu的tensorflow。在

检查这个link看看哪个版本需要什么Cuda和Cudnn

我建议您卸载tensorflow的CPU和GPU版本,然后重新安装。在

如果你想保持你的cuda版本为9,那么安装tensorflow版本1.5。如果你使用水蟒安装它很可能你会得到版本1.4,它只支持CUDA8

相关问题 更多 >