Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU
如上图报错,说gpu没有支持的kernel。原因是应为在tensorflow中,定义在图中的op,有的只能再cup中运行,gpu中不支持。解决方法就是让op自动识别,让它选择在合适的地方运行即可。如果op中有标识的话,在运行的时候指定在cup上执行,如果无法区分的话,可以试试在sess.run的时候加入allow_soft_placement=True。例如:sess = tf.Session(c...
转载
2018-12-28 15:23:19 ·
376 阅读 ·
0 评论