安装bert4keras时要注意的问题

问题

进入自己的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

成功!

  • 4
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值