最近在实习,老板说让我把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的链接:
关于tensorflow2.4.0配置GPU的那些事
最新推荐文章于 2025-03-17 21:48:39 发布