已安装PyTorch但调用BertModel时出现导入错误求助

问题症状 #

已安装并导入PyTorch，但执行以下代码时：

model = BertModel.from_pretrained("Rostlab/prot_bert")

出现报错：

ImportError: BertModel requires the PyTorch library but it was not found in your environment. However, we were able to find a TensorFlow installation. TensorFlow classes begin with "TF", but are otherwise identical.

解决方法 #

1. 确保导入PyTorch版本的BertModel #

环境中同时存在TensorFlow时，transformers可能默认加载TensorFlow实现的模型。需明确导入PyTorch版本的BertModel：

# 标准导入方式
from transformers import BertModel

# 若仍有问题，直接指定PyTorch实现路径导入
from transformers.models.bert.modeling_bert import BertModel

2. 验证当前环境的PyTorch是否正常可用 #

运行以下代码确认PyTorch已正确安装在当前运行环境：

import torch
print(torch.__version__)

若报错，说明当前环境未正确安装PyTorch，需重新在该环境中安装对应版本的PyTorch。

3. 更新transformers库至最新版本 #

旧版本的transformers可能存在后端识别问题，执行命令更新：

pip install --upgrade transformers

4. 强制指定使用PyTorch后端 #

在代码开头添加环境变量配置，强制transformers使用PyTorch：

import os
os.environ["USE_TORCH"] = "1"

# 之后再导入模型
from transformers import BertModel
model = BertModel.from_pretrained("Rostlab/prot_bert")

内容的提问来源于stack exchange，提问作者suraj preetham