tensorflow-gpu2.x无法检测到GPU设备
tensorflow-gpu2.x无法检测到GPU设备
1.检查
在解决问题前检查一下tensorflow是否只能检查到CPU,而无法检查到GPU,而非无法调用GPU。
import tensorflow as tf
print(tf.test.is_gpu_available())
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')
print(gpus, cpus)
#打印结果为
#False
#[] [PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
2.解决方法
2.1问题一:没安装对应的cudnn和cudatoolkit依赖
解决办法:安装对应的cudnn和cudatoolkit
这个我就不重述了我博文里面有:https://blog.csdn.net/qq_43203949/article/details/108479784.
注意我的安装环境是:其中Driver Version: 441.83 NVIDIA-SMI 441.83没有绝对的要求。
2.2问题二:安装了对应的依赖却缺失了关键文件(cudnn64_7.dll)
如发生一下问题:(在pycharm的红色字体报告中可以看到)
Could not load dynamic library 'cudnn64_7.dll'; dlerror: cudnn64_7.dll not found
解决办法:去英伟达那里下载对应的cudnn文件,并将缺失文件拷贝到对应的文件夹中即可。
这个我就不重述了,我这个也写了一个博文:
https://blog.csdn.net/qq_43203949/article/details/108550960.
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)