faster-rcnn-tf2 icon indicating copy to clipboard operation
faster-rcnn-tf2 copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

运行predict.py时报错Could not find valid device for node

Open Baymaaax opened this issue 4 years ago • 7 comments

tensorflow.python.framework.errors_impl.NotFoundError: Could not find valid device for node. Node:{{node NonMaxSuppressionV3}} All kernels registered for op NonMaxSuppressionV3 : device='CPU'; T in [DT_FLOAT] device='CPU'; T in [DT_HALF] [Op:NonMaxSuppressionV3]

Baymaaax avatar Dec 01 '20 12:12 Baymaaax

你是CPU版本的吗

bubbliiiing avatar Dec 07 '20 07:12 bubbliiiing

大概率是环境配置的问题

bubbliiiing avatar Dec 07 '20 07:12 bubbliiiing

tensorflow版本的问题，以解决，谢谢回复

牛宇童

在 2020年12月7日，15:01，Bubbliiiing [email protected] 写道：

 你是CPU版本的吗

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Baymaaax avatar Dec 07 '20 07:12 Baymaaax

我也遇到了一模一样的问题，我是gpu，tf2.1版本的，请问您是怎么解决的呢？

hamster-tea avatar May 17 '21 22:05 hamster-tea

应该是tensorflow、cuda和cudnn不匹配吧

bubbliiiing avatar May 21 '21 14:05 bubbliiiing

应该是tensorflow、cuda和cudnn不匹配吧

我解决了，不是不匹配的原因，就是tf版本低了，我重新装了一个2.2版本的就能跑了。

hamster-tea avatar May 21 '21 14:05 hamster-tea

好的

bubbliiiing avatar May 21 '21 14:05 bubbliiiing