加载Roberta基模型的问题

Question

我试图使用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

有关更多详细信息，请参阅这里。