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

已于 2022-11-10 14:38:48 修改
阅读量1.7k 收藏 8
点赞数 10
CC 4.0 BY-SA版权
分类专栏： macOS Jupyter Notebook 文章标签： cnn
于 2022-10-09 23:13:33 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/weixin_45041379/article/details/127227546
本文记录了使用M1Pro芯片的Macbook Pro运行PyTorch模型LeNet与AlexNet的过程。通过设置设备为mps来利用GPU资源，观察到GPU占用率逐渐升高，而CPU占用稳定。此外还分享了模型训练期间的系统资源使用情况。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

记录M1 Pro芯片Macbook Pro mps跑模型

设置pytorch的模式

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

device = torch.device("mps")

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

LeNet

在这里插入图片描述

AlexNet

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

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

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

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

评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值