Tensorflow Object Detection API 版本对应关系(CUDA/cudnn/tensorflow-gpu踩坑)

本文详细介绍了在使用TensorFlow Object Detection API时,正确配置tensorflow-gpu、CUDA和cuDNN版本的重要性。强调了tensorflow-gpu的版本不得低于1.12.0,以及CUDA和cuDNN的具体版本要求,避免模型运行失败。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

关于Tensorflow Object Detection API中tensorflow-gpu的版本选择:

记得一定要去看官网的要求!

官网是这么说的:它的版本是tensorflow-gpu 1.12.0,你不能低于1.12.0版本,不然model zoo里面的模型你跑不了~

官网:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

好好看看这段话:

所以,你版本可要装对了~

再来,CUDA和cudnn的版本和tensorflow-gpu的对应关系(我另一篇文章也有所提及),详见官网:
https://tensorflow.google.cn/install/source

一直往下拉:
在这里插入图片描述

看到没有,这说明你要装的必须是CUDA9.0的,我装了9.2的还不行~

后来换成版本9.0.176

然后cudnn版本要7.2.1以上的才不会出问题,因为我运行的faster-cnn报错说模型是7.2.1编译的,cudnn必须大于等于这个版本!!!

所以最终的版本是
tensorflow-gpu:1.12.0
CUDA:9.0.176
cudnn:7.4.2.24

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

龚大龙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值