RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
时间: 2023-10-22 22:04:07 浏览: 158
Python RuntimeError: thread.__init__() not called解决方法
这个错误通常是由于cuDNN库与CUDA版本不兼容引起的。建议您检查CUDA和cuDNN版本是否匹配。如果版本匹配,请尝试重新安装CUDA和cuDNN。如果问题仍然存在,您可以考虑降低cuDNN版本或升级CUDA版本。另外,您还可以尝试在代码中添加以下代码,以便在运行时打印出更详细的错误信息:
```python
import torch
torch.backends.cudnn.enabled = True
torch.backends.cudnn.benchmark = True
torch.backends.cudnn.deterministic = True
# your code here
```
这将启用cudNN加速,并打印出更详细的错误信息,以帮助您进一步诊断问题。
阅读全文