安装bert4keras时要注意的问题

安装bert4keras时要注意的问题,第1张

问题

进入自己的conda环境后,自己的安装顺序:

pip install tensorflow-gpu=2.6.0
pip install keras=2.6.0
pip install bert4keras

在pycharm里写完代码之后运行,会出现各种has no attribute错误,后来发现是版本tensorflow和keras不一致?问题就出现在这里,安装bert4keras的时候,它会自动重新安装keras(今天是2022.4.16,自动重新安装的版本是keras2.3.1),所以keras2.3.1和tensorflow2.6.0不一致。

解决

先查看与keras2.3.1匹配的tensorflow版本,发现2.2.0匹配
所以

pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
conda install cudnn=7.6
pip install bert4keras

成功!

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/716249.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-25
下一篇 2022-04-25

发表评论

登录后才能评论

评论列表(0条)

保存