运行时错误：cuDNN 版本不兼容

Question

我在 google colab 中用 PyTorch 写了一个 LSTM NLP 分类器，它运行良好。现在，我在 google colab 上运行它临，但我收到此错误：

RuntimeError：cuDNN 版本不兼容：PyTorch 是针对 (8, 3, 2) 编译的，但发现运行时版本 (8, 0, 5)。 PyTorch 已经与 cuDNN 捆绑在一起。解决此错误的一种选择是确保 PyTorch 可以找到捆绑的 cuDNN。一种可能是 LD_LIBRARY_PATH 中存在冲突的 cuDNN。

我不知道如何解决这个问题。我在 colab pro 上使用 GPU。 我试过这个link，但没有用。 我如何声明设备：

DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

Answer 1

通过将 cuDNN 升级到 8.4 修复

参考 (https://github.com/JaidedAI/EasyOCR/issues/716)

如果你使用的是 google colab uae 这个命令

!pip install --upgrade torch torchvision