我安装并导入了PyTorch。但是，当我使用BertModel时，我仍然收到这个导入错误。我需要一些帮助。
我得到的错误是：
导入错误：BertModel需要PyTorch库，但在您的环境中找不到该库。不过，我们能够找到TensorFlow安装。TensorFlow类以"TF"开头，但在其他方面相同
我试过这个：

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

我得到了这个：