用PyTorch训练一个小型的神经网络

安装环境

PyTorch 需要自行根据自己的英伟达GPU型号，选择CUDA对应的版本进行安装，安装方法自行搜索 python 3.11

运行

python run.py

训练目标

识别各种手写体的数字 0-9