如果出现:
No registered '_CopyFromGpuToHost' OpKernel for CPU devices compatible with node
tensorflow.python.framework.errors_impl.NotFoundError: No registered '_CopyFromGpuToHost' OpKernel for CPU devices compatible with node swap_out_gradients/bert/encoder/layer_0/attention/output/dense