基于PyTorch，在PC上完成MNIST手写数字识别模型训练并生成Onnx文件
基于PyTorch，在PC上完成MNIST手写数字识别模型训练并生成Onnx文件
3月1日修改
注意！网上有很多版本无法实现手写数字模型训练，我这个亲手实验，是可以成功的！
这个版本是在windows环境下完成手写数字模型训练。
首先学习下PyTorch：
中文版pytorch学习手册：
1.安装环境
1.1安装CUDA
首先查看CUDA环境：
cmd命令行内输入nvidia-smi.exe，查看本机显卡驱动版本及可用的CUDA版本。
附件不支持打印
加载失败，