赞
踩
直接用了github给的bert模型和数据,想先跑起来一个,但是本身环境里的tensorflow版本不对应
TensorFlow报错:AttributeError: module ‘tensorflow_core._api.v2.train’ has no attribute 'Optimizer‘
原因:
TensorFlow版本更新到2.0,Optimizer方法调用方式改变 。
将tf.train.Optimizer,更改为tf.optimizers.Optimizer即可。
flag报错,再改:
flags = tf.flags
FLAGS = flags.FLAGS
以上原文链接:https://blog.csdn.net/weixin_42720673/article/details/103066349
tensorflow下载版本要和cuda版本对应(在别处看的,不知道是不是必须,还是谨慎为好),先查看当前tensorflow及相关版本:conda list
查看服务器cuda版本:
cd /usr/local
ls
按照cuda版本去下载对应tensorflow:
pip3 install tensorflow-gpu==1.13.1 -i https://pypi.douban.com/simple
接着报错:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory Failed to load the native TensorFlow runtime.
点击加号添加新的环境变量,LD_LIBRARY_PATH=/usr/local/cuda/lib64
无报错,可以正常运行!!!!!开心!!!!!!!!!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。