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设置小一点。