使用keras时解决 Failed to get convolution algorithm. 报错问题
解决 Failed to get convolution algorithm. 报错问题
小白记录一下在深度学习问题上的报错代码,希望也能给予有相同问题的人一些帮助。
报错场景:
在训练完成了一个深度神经网络的前提下,希望输出中间层特征输出结果,于是使用Model构建了一个新的网络,原始的inputs是输入,model.get_layer().output提取的层是输出。对新网络使用model.predict()指令的时候报了如下错误。
报错代码:
E tensorflow/stream_executor
原创
2020-09-12 17:12:39 ·
598 阅读 ·
0 评论