RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
Posted AI浩
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR相关的知识，希望对你有一定的参考价值。
params = {
"model": "UNet11",
"device": "cuda",
"lr": 0.001,
"batch_size": 16,
"num_workers": 4,
"epochs": 10,
}
model = create_model(params)将num_workers改为0
如果还是没有解决，试试将batchsize设置小一点。
以上是关于RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR的主要内容，如果未能解决你的问题，请参考以下文章
RuntimeError：cuDNN 错误：CUDNN_STATUS_NOT_INITIALIZED 使用 pytorch
RuntimeError: cudnn RNN backward can only be called in training mode
无法创建 cudnn 句柄：CUDNN_STATUS_INTERNAL_ERROR
无法创建 cudnn 句柄：CUDNN STATUS INTERNAL ERROR