我用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')
1条答案
按热度按时间js4nwp541#
通过将cuDNN升级到8.4修复
引用(https://github.com/JaidedAI/EasyOCR/issues/716)
如果您使用的是google colab uae,则执行此命令