进入自己的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
成功!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)