栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

tensorflow无法使用gpu(tensorflow检测不到gpu)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

tensorflow无法使用gpu(tensorflow检测不到gpu)

最近几天配置PT和TF的环境遇到了太多坑了,首先按照网上的教程整合安装好Anaconda以及PT和TF的GPU版本之后,按照下图进行测试:

Pytorch测试方法

Tensorflow测试方法
都分别遇到了返回False
以及无法找到dll文件的问题(文件在对应目录,cudnn等版本也是正确的)
之后想到去设备管理器查看NVIDIA显卡驱动的版本时发现:

当时设备状态中显示Windows无法验证此设备所需的驱动程序的数字签名…代码52,
并且此时的驱动资源状态为不可用,顿悟,原来自己之前将CUDA驱动更改为10.0版本导致系统不能合法识别该驱动签名
因此需要如下图所示依次解决该强制签名问题:









然后重启电脑 , 打开电脑 设备管理–> 在线更新驱动就好了

此时显卡状态正常:

然后再按照之前的步骤分别切换到对应的环境进行GPU版测试
结果分别如下:

在测试Tensorflow-GPU时要注意:如果在pycharm中用上述方法进行测试会报找不到dll文件的错
需要在对应环境的命令行Python环境中依次输入并执行以下命令:

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))


得到上图所示结果证明所有GPU版本安装并测试完成!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/772804.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号