加载Roberta基本模型时出现问题

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

请参阅此处了解更多详细信息。