关于tensorflow2.4.0配置GPU的那些事

最近在实习,老板说让我把tensorflow2.4.0的gpu给配置好,方便我训练,就这,我都配置了大概四五次,第五次成功了,记录一下前几次的经历:
看网上各种参差不齐的教程,说用cudn10.0版本的成功了,于是乎,我就直接装了cuda10.0以及与之对应的cudnn,但是装上去发现并没有什么用,训练的时候总是会出现某一个.dll找不到。这其实就是没有配置好的原因。下面来说说我最终配置tensorflow2.4 +cuda11.0 +cudnn 8.0.5版本的
首先第一步肯定是要看你的电脑是否支持英伟达的显卡,并确定你的英伟达的显卡的驱动支持的cuda在11.0版本以上,可以通过命令行查看:nvidia-smi
在这里插入图片描述
因为tensorflow2.4版本目前需要cuda11.0以上的支持,我也不知道为什么。如果不支持,一般360驱动助手可以自动更新到最新版本,对了 ,好像还要求英伟达的显卡的算力在3.5以上,我用的是GTX960的,算力是5.2,可以,关于算力的查看可以自行百度,一百度就出来了。
然后知道了你的电脑显卡驱动支持就可以进行我们的文件下载准备了
我这里直接给大家放了百度网盘的链接,大家自己取:
这个是cuda11.0的链接
链接:https://pan.baidu.com/s/1olNJfBnTDrw_7S3NYnd0sw
提取码:nump
这个是cudnn 8.0.5的链接:

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值