出错 InternalError: Blas GEMM launch failed : a.shape=(64, 784), b.shape=(784, 32), m=64, n=32, k=784

错误代码显示

// 代码如下所示。主要是tensorflow2.0-GPU训练mnist数据集时，出错：InternalError:  Blas GEMM launch failed : a.shape=(64, 784), b.shape=(784, 32), m=64, n=32, k=784

// An highlighted block
var foo = 'bar';
# 模型的训练、验证和测试与训练模型完全相同
# 下面使用mnist数据集进行展示
(x_train, y_train), (x_test, y_test) = keras.datasets