錯誤時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. 實現它自己。