使用PyTorch与Transformers微调Gatortron-BERT时遇NumPy初始化警告

你遇到的Failed to initialize NumPy: _ARRAY_API not found警告，核心原因是NumPy 1.19.5版本与Python 3.12不兼容。Python 3.12引入的_ARRAY_API相关接口在老旧的NumPy版本中不存在，导致PyTorch调用NumPy时触发警告。

具体解决步骤： #

• 升级NumPy到兼容版本：执行以下命令将NumPy升级到支持Python 3.12的版本（建议1.26.0及以上）：

  pip install --upgrade numpy
  

• 同步升级PyTorch（可选但推荐）：确保PyTorch版本与新NumPy、Python 3.12兼容，避免后续出现其他兼容性问题：

  pip install --upgrade torch
  

• 验证修复：重新运行你的代码，该警告应该会消失。

额外提示： #

你的代码中使用的BertModel是BERT的基础模型，本身不会输出任务相关的logits。如果是做下游任务（如分类、问答等），建议替换为对应任务的模型类（例如BertForSequenceClassification、BertForQuestionAnswering），否则当前代码中对outputs[0]做argmax得到的结果可能不符合预期。

内容的提问来源于stack exchange，提问作者Shaikh Shehbaz