使用Colab GPU时出错，而没有CPU时出错

Question

我正在Google Colab中尝试一些代码。 使用CPU可以正常工作，但是当我切换到GPU时会显示错误。

自包含代码：

list

错误：

>>> print(list(mapping[arr]))
[0, 10, 30, 55, 55, 66, 70, 90, 80, 90, 30, 20, 40, 66]

为什么使用CPU和GPU之间会有这种区别？

Answer 1

这可能与分别在CPU和GPU内核上的tensorflow和tensorflow-gpu软件包有关。

您可以绕过它，但是从BatchNormalization layer移除axis = 1

更改：

X = BatchNormalization(axis = 1, momentum=BATCH_NORM_MOMENTUM, epsilon = BATCH_NORM_EPS, name = step+'b1_bn_a')(X)

收件人：

X = BatchNormalization(momentum=BATCH_NORM_MOMENTUM, epsilon = BATCH_NORM_EPS, name = step+'b1_bn_a')(X)