BERT 运行MRPC demo问题

最新推荐文章于 2024-11-28 12:15:37 发布

刘大鸭

最新推荐文章于 2024-11-28 12:15:37 发布

阅读量777

点赞数

分类专栏： BERT 文章标签：自然语言处理

本文链接：https://blog.youkuaiyun.com/qq_36642243/article/details/109671703

版权

在尝试运行BERT处理MRPC任务时遇到问题。错误可能由于tensorflow-gpu版本不匹配导致，解决方法是降级到1.12.0版本。另外，当GPU使用率过高引发崩溃时，建议降低batch_size并关闭不必要的程序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

如果出现：

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