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

作者：花生_TL007 | 2024-04-10 22:54:02

踩

m1 跑模型

设置pytorch的模式

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

device = torch.device("mps")
1

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

在这里插入图片描述

网络结构：
在这里插入图片描述

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

在这里插入图片描述
gpu的占比从70+、80+到90+，cpu占比20+，不会再往上。不是很懂芯片设计。

跑完了，风扇声也停了。
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/401466