Tensorflow，CNN：使用GPU时，我是否必须使用'NCHW'格式制作卷积神经网络？

Question

我查看了DQN实施代码（https://github.com/devsisters/DQN-tensorflow），并在使用CNN时发现此代码生成NCHW格式为GPU的网络，否则为NHWC。

我想知道这是否具有可取性......我的意思是，即使我使用GPU，也可以用CNN格式生成NHWC网络吗？

Answer 1

如果你不太关心表现，那可能没问题。通常，在GPU上，NCHW更快。

为什么要在GPU上专门使用NHWC？ 你可以在它们之间进行转换，参见 Convert between NHWC and NCHW in TensorFlow