MXNet(Gluon)指定GPU(0)上下文仍用CPU问题及后续验证

之前我碰到个头疼的问题：明明在MXNet(Gluon)里指定了GPU(0)作为运行上下文，但实际训练全程都在用CPU。经过一番折腾，现在终于把事情捋清楚了：

后续测试与关键发现 #

• 我自己编写了一套代码，把数据存在本地，还简化了精度计算的逻辑，跑起来速度直接上来了，GPU也正常工作了
• 试了所有在MXNet里搭建的CNN模型——哪怕是MNIST那种超简单的小模型，GPU性能都比CPU强一大截，说明框架本身的GPU加速机制是没问题的
• 原来的问题根本不是框架的锅，是当时用的教程代码存在特定情况导致的，所以那事儿已经不算真实的框架问题了

当前进展 #

现在我在跑Gluon多层感知器MNIST的教程代码，只修改了上下文设置为GPU(0)，GPU运行完全正常，训练效率符合预期

内容的提问来源于stack exchange，提问作者Tristan HB