pytorch 运行时错误:cuDNN版本不兼容

rt4zxlrg  于 2022-12-13  发布在  其他
关注(0)|答案(1)|浏览(393)

我用PyTorch写了一个LSTM NLP分类器,在google colab中运行得很好。现在,我在google colabpro上运行它,但是我得到这个错误:
运行时错误:cuDNN版本不兼容:PyTorch是针对(8,3,2)编译的,但找到的是运行时版本(8,0,5)。PyTorch已经捆绑了cuDNN。解决此错误的一个方法是确保PyTorch可以找到捆绑的cuDNN.one。可能是LD_LIBRARY_PATH中存在冲突的cuDNN。
我不知道如何解决这个问题。我在colab pro上使用GPU。我试过这个link,它不工作。我如何声明设备:

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

js4nwp54

js4nwp541#

通过将cuDNN升级到8.4修复
引用(https://github.com/JaidedAI/EasyOCR/issues/716

如果您使用的是google colab uae,则执行此命令

!pip install --upgrade torch torchvision

相关问题