zl程序教程

您现在的位置是:首页 >  其它

当前栏目

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

Error status internal cudnn RuntimeError
2023-09-14 09:05:44 时间
params = {
    "model": "UNet11",
    "device": "cuda",
    "lr": 0.001,
    "batch_size": 16,
    "num_workers": 4,
    "epochs": 10,
}
model = create_model(params)

将num_workers改为0

如果还是没有解决,试试将batchsize设置小一点。