记录｜M1 Pro mps跑模型LeNet和AlexNet

设置pytorch的模式

设置mps才会用到gpu，一般Nvidia的那种方法没用，还是run on cpu

device = torch.device("mps")

然后把device参数传进网络模型

LeNet

AlexNet

网络结构：

经历了很长一段时间才开始画图。。
风扇从过了4个epoch开始叫得有点可怕
电量也开始掉
其实看网络的情况，是比当时毕设跑的那个output10000+的好很多的
内存压力也不大，主要都被我开着的虚拟机给占了

gpu的占比从70+、80+到90+，cpu占比20+，不会再往上。不是很懂芯片设计。

跑完了，风扇声也停了。