加载基于 Roberta 的模型时出现问题

Question

我正在尝试使用 AutoTokenizer.from_pretrained('roberta-base') 的 Roberta-base 模型，但出现以下错误：

RuntimeError: Failed to import transformers.modeling_tf_utils because of the following error (look up to see its traceback):
No module named 'tensorflow.python.keras.engine.keras_tensor'

我尝试安装 tensorflow 但仍然出现同样的错误。知道发生了什么吗？

Answer 1

你需要安装GPU，试试这个：

pip install --ignore-installed --upgrade tensorflow-gpu

更多详情请见here。