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

关于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

猜你喜欢

转载自blog.csdn.net/m0_37970224/article/details/89299014