PyTorch 已安装。 但是，当我使用 BertModel 时，出现以下导入错误：

Question

I installed and imported PyTorch . Still, I am getting this import error, while I'm using BertModel . I need some help here.

Error I'm getting is this:

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 identica

I tried this:

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

I'm getting this:

Answer 1

I encountered the same error but it has disappeared after restarting jupyter.