在MNIST上进行训练，可以说是计算机视觉里的“Hello World”任务了。 而如果使用PyTorch的标准代码训练CNN，一般需要3分钟左右。 但现在，在一台 笔记本电脑上就能将时间缩短200多倍。 这是一台装有GeForce GTX 1660 Ti GPU的笔记本。 我们需要的还有Python3.x和Pytorch 1.8。

导语：对于图像分类任务，当前最先进的架构是卷积神经网络 (CNNs)，无论是面部识别、自动驾驶还是目标检测，CNN 都得到广泛使用。 雷锋网 AI 研习社按：本文为雷锋网字幕组编译的技术博客，原标题 A simple 2D CNN for MNIST digit recognition，作者为 Sambit Mahapatra。