没有支持GPU设备的内核可用于SparseTensorDenseMatMul_grad

时间:2016-07-21 09:54:06

标签: tensorflow

我在图表中使用 tf.sparse_tensor_dense_matmul 操作构建模型时遇到了问题。部分错误信息粘贴如下,

这是否意味着没有GPU内核支持来计算“ SparseTensorDenseMatMul_grad ”的渐变?我可以在会话配置中使用“ allow_soft_placement = Ture ”成功构建模型。但是,由于某些特殊原因,我需要在GPU上保留所有计算。有谁知道如何解决这个问题?或者我需要自己实现这个操作系统的CUDA内核?非常感谢。

tensorflow.python.framework.errors.InvalidArgumentError: Cannot assign a device to node 'gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
         [[Node: gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1 = Slice[Index=DT_INT32, T=DT_INT64, _device="/device:GPU:0"](Placeholder_2, gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1/begin, gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1/size)]]
Caused by op u'gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1', defined at:

0 个答案:

没有答案