在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. 为自己实施。