📌  相关文章
📜  无法创建 cudnn 句柄:CUDNN_STATUS_INTERNAL_ERROR (1)

📅  最后修改于: 2023-12-03 15:26:15.820000             🧑  作者: Mango

无法创建Cudnn句柄:CUDNN_STATUS_INTERNAL_ERROR

当您在运行深度学习模型时收到 "无法创建 cudnn 句柄:CUDNN_STATUS_INTERNAL_ERROR " 的错误消息时,这意味着您的程序无法创建一个 cudnn 句柄,这是在进行卷积操作时使用的一个对象。这个错误通常与以下几个问题有关:

  1. CUDNN 安装问题: 在安装 cudnn 时可能会出现问题,比如不正确地进行了安装,或者安装的版本与您的 CUDA 版本不匹配。您可以重新安装 cudnn 并确保安装程序和 CUDA 版本匹配。

  2. 显存问题: 此错误可能是因为显存不足所致。您可以尝试降低摆放在显存中的张量大小及数目,或者使用更高容量的 GPU。

  3. 操作错误: 如果您在进行卷积操作时,输入数据、卷积核或输出数据中的任何一个大小或形状不正确,都可能会导致此问题。确保您的代码正确。

以下是可以尝试的解决方案:

  • 更新 CUDNN 版本,保证与 CUDA 版本匹配。
  • 检查显存是否不足,尝试降低计算的张量大小和数量。
  • 检查代码错误和输入数据的正确性。
  • 若已使用 pyTorch,则尝试:torch.cuda.empty_cache() 和 cuDNN_benchmark = True

希望这些解决方案能够帮助你解决 "无法创建 cudnn 句柄:CUDNN_STATUS_INTERNAL_ERROR " 的问题。