win10上针对tensorflow目标检测模型进行GPU加速时遇到的问题

使用tensorflow-gpu进行训练时遇到的错误:

tensorflow.python.framework.errors_impl.UnknownError

或

AttributeError: module 'tensorflow' has no attribute 'contrib'   （tensorflow2.0以后已经没有contrib这个属性了）

我可以99%确定是你的tensorflow版本以及CUDA和cudnn版本不匹配造成的，如果你想把所有的CUDA和cudnn版本都试验一遍的话，我相信1个月可能也不够。

你只需要安装如下的进行尝试即可：

• cuda 9.0 
• tensorflow-gpu 1.12.0
• cudnn 7.4.1.5

也就是说，安装了以上3个，在以下2个链接的模型中都能用GPU模式跑起来，在代码中不需要设定什么GPU，自动就用GPU跑起来了。

https://github.com/YunYang1994/tensorflow-yolov3

https://github.com/tensorflow/models/tree/master/research/object_detection

来源：oschina

链接：https://my.oschina.net/dapsjj/blog/3156043