keras CUDNN_STATUS_BAD_PARAM core dump 问题

使用keras训练模型时遇到 CUDNN_STATUS_BAD_PARAM core dump 问题，发现时因为我使用的是GPU并行方式来训练模型，最后一个batch的样本如果没有大于等于GPU的个数的话，就会发生这个问题。我解决的方法是，设置合适的batch_size，让最后一个batch的样本个数大于等于GPU的个数。因为并行模式里，每个GPU至少要有一个样本来跑。