加载 Roberta-base 模型时出现问题

Question

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

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。