首先说一下我遇到这个问题时的环境:在使用bert训练模型时,会出现如下提示:
错误提示
也有很多网友也遇到了这样的问题,以下是github上讨论区里面的话题:
github上讨论话题
在网上查询该问题的时候很多人说是tensorflow版本问题引起的,tensorflow2.x和tensorflow1.x内容相差较大,而‘Optimizer’需要tensorflow1.x,我当前环境下的tensorflow是2.4.1的,于是很自然地运行:

pip uninstall tensorflow
pip install tensorflow==1.15

(版本随便选择了1.15,因为看到GitHub上有人说1.15可以)
(如果使用的是GPU版本的,请将上述代码中的tensorflow换成tensorflow-gpu)
然而,却出现了以下提示:
无法安装1.15版本
划重点:最终解决方法!!!
python版本过高,需要更换至更低版本的python,在我将python从3.8换至3.6的时候,成功安装tensorflow1.15版本,并且bert运行一路畅通无阻。
成功安装了tensorflow1.15

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐