在跑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版本相对应
点击查看官网现在对应版本