【Pytorch报错】RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

本文主要是介绍【Pytorch报错】RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

报错信息：

File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 297, in _conv_forwardreturn F.conv1d(input, weight, bias, self.stride,
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)

解决方法：
禁用cuDNN，在训练代码的顶部添加以下代码：

import torch
torch.backends.cudnn.enabled = False

其他说明：
torch.backends.cudnn.benchmark = False 是关闭自动调优。
torch.backends.cudnn.enabled = False 是完全禁用cuDNN。
在遇到cuDNN相关的问题时，首先尝试关闭自动调优，如果问题依然存在，再考虑完全禁用cuDNN。

这篇关于【Pytorch报错】RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---