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

安装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运行程序了
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值