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

2021-10-28

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

2021-10-28

安装tensorflow-gpu中遇到的部分问题

本人将对自己遇到的问题进行总结
首先说一下个人配置,我用的是笔记本,cpu是i5-10300H,gpu是rtx2060。
我开始安装的tensorflow版本以及配套软件,cuda11.0.2,cudnn8.0.5,tensorflow-gpu2.4.0,python3.8.12。
安装完之后在pycharm中运行程序
我的测试代码是

import tensorflow as tf
print(tf.version)
print(1, tf.config.list_physical_devices('GPU'))

之后出现了下面的问题

程序说是没有用gpu运行。
在网上查了查,大部分说是显卡驱动或者cuda,cudnn与tensorflow版本不匹配,下面给出了各个版本的对应关系
显卡驱动与cuda的对应关系
tensorflow-gpu和cuda的对应关系
所以我尝试显卡所对应的最新的驱动版本,可以通过这个查询显卡所对应的驱动版本。
最后,更新完,发现还是出现这个错误。我又看了看安装过程一个个排查,发现问题出现在安装cudnn的过程,我是直接将cudnn中的文件夹复制到cuda中对应的文件夹下,这是错误的,应该直接将cudnn中的文件夹中的文件复制到cuda中对应的文件夹中。
不是复制这个

是复制这几个

改完之后,变成这样,能用gpu运行程序了

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

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

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