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

Question

我正在尝试使用基于 Roberta 的模型，AutoTokenizer.from_pretrained('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

有关更多详细信息，请参见此处。