在跑Bert源码的时候发现出了个小bug
请添加图片描述
连续报两个Warning

  • The name tf.train.Optimizer is deprecated. Please use
    tf.compat.v1.train.Optimizer instead.
  • The name tf.app.run is deprecated. Please use tf.compat.v1.app.run
    instead.

由于我用的是TensorFlow1.15.0和CUDA10.0的版本,想着是不是版本不合适,所以将其卸载,并移除虚拟环境。
请添加图片描述

pip uninstall tensorflow

请添加图片描述
别忘了卸载干净
在这里插入图片描述

接下来重新安装TensorFlow低版本即可

conda create -n tensorflow pip python=3.7
activate tensorflow
conda install tensorflow=1.13.1
//gpu 用下面这个
conda install tensorflow-gpu=1.13.1

注意TensorFlow,CUDA和Python版本相对应
请添加图片描述
点击查看官网现在对应版本

参考:https://github.com/keras-team/keras/issues/13004

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐