错误时GPU上运行tf.SVD - tensorflow

Question

我试图运行在GPU上采用tf.SVD代码，但它给了我这个错误：

No registered 'Svd' OpKernel for GPU devices compatible with node Svd = Svd[T=DT_FLOAT, compute_uv=true, full_matrices=false](x) 
. Registered: device='CPU'; T in [DT_FLOAT] 
    device='CPU'; T in [DT_DOUBLE] 
    device='CPU'; T in [DT_COMPLEX64] 
    device='CPU'; T in [DT_COMPLEX128] 

不知道如何解决这个问题？

Answer 1

该错误告诉您此操作不适用于GPU。所以，你的选择是：

1. 坐，等到有人会实现它
2. 使用CPU版本
3. 实现它自己。