PyTorch 已安裝。 但是，當我使用 BertModel 時，出現以下導入錯誤：

Question

我安裝並導入PyTorch 。 不過，我在使用BertModel時遇到了這個導入錯誤。 我需要一些幫助。

我得到的錯誤是：

ImportError: BertModel 需要 PyTorch 庫，但在您的環境中找不到。 但是，我們能夠找到 TensorFlow 安裝。 TensorFlow 類以“TF”開頭，但其他方面相同

我試過這個：

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

我明白了：

Answer 1

我遇到了同樣的錯誤，但是重啟jupyter后它就消失了。