bojone / bert4keras

keras implement of transformers for humans
https://kexue.fm/archives/6915
Apache License 2.0
5.37k stars 929 forks source link

去掉setup.py中kerase的限制 #417

Open cole-dda opened 3 years ago

cole-dda commented 3 years ago

提问时请尽可能提供如下信息:

pip 21.3 会自动Backtracking,当使用最新的tensorflow2.7的时候,需要keras2.7

而bert4keras限制如下

screenshot_4791

是否可以去掉这个限制? 导致pip安装容易冲突

bojone commented 3 years ago

这是不可以的。

只有keras <= 2.3.1才能称为keras,其他都是披着keras皮的tf.keras,两者是不等价的;如果用tf.keras,请直接通过TF_KERAS=1来调用,不要当作keras来用。

就算去掉限制让你装上,大概率也在运行过程中报错。

另外,对新显卡(比如A100)支持最好的tf依然是tf 1.15,所以目前我还没找到任何使用tf2的必要性,因此当前做法完全够用。珍惜生命,远离tf2。

参考:https://developer.nvidia.com/zh-cn/blog/accelerating-tensorflow-on-a100-gpus/