TensorFlow2.4可以在MacBook Pro/Mac Pro上利用GPU做机器学习训练了

最新推荐文章于 2025-02-24 15:12:57 发布

thl789

最新推荐文章于 2025-02-24 15:12:57 发布

阅读量6.8k

点赞数 5

分类专栏： ai 文章标签： MacBook Pro Mac Pro TensorFlow2.4 机器学习 Training

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/thl789/article/details/109911959

版权

ai 专栏收录该内容

23 篇文章

订阅专栏

在Mac上使用TensorFlow进行机器学习训练时，从TF2.4版本开始，无论是Intel架构还是M1芯片的MacBookPro，都能通过GPU显著提高训练速度。M1架构相较于仅使用CPU的训练性能提升了7倍。

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

田海立@CSDN 2020-11-21

之前MacBook上TensorFlow只能利用CPU做训练，TF2.4开始可以利用GPU做训练了，并且不管是M1的MacBook Pro，还是Intel架构的MacBook Pro还是Mac Pro都是支持的。Apple披露这一信息，并且po出性能对比数据。

MacBook Pro上利用GPU训练

看下面的性能数据，对比了CPU数据、Intel架构的MacBook Pro以及M1架构的MacBook Pro：

Shows a chart that compares three models. One that uses half a million utterances, another that uses one million utterances, and a third that uses five million utterances. The accuracy increases with the number of utterances. The three accuracies are 99 point forty six percent, 99 point 62 percent, and 99 point 85 percent.

结果显示：M1架构的训练性能比CPU提升了7倍；Intel架构的没那么明显。

其中的机器及软件配置：

CPU是13-inch Intel架构的Macbook Pro，跑的是TF2.3
Intel架构的GPU加速机器以及M1芯片的GPU加速机器跑的是TF2.4 prerelease
Intel架构的13-inch Macbook Pro的配置：1.7GHz 4核 i7 CPU + Intel Iris Plus Graphics 645 GPU + 16GB内存 + 2TB SSD硬盘
M1芯片的13-inch Macbook Pro的配置：M1（4核高性能+4核高效能CPU + 8核GPU + 16核Neural Engine）+ 16GB内存 + 256GB SSD硬盘

只是，M1里有NPU，这个有利用NPU吗，还是仅仅GPU？Apple没过多披露，字里行间也只提到了GPU，保留关注。

Mac Pro上利用GPU训练

Mac Pro上CPU与GPU训练的数据如下：

Shows a chart that compares three models. One that uses half a million utterances, another that uses one million utterances, and a third that uses five million utterances. The accuracy increases with the number of utterances. The three accuracies are 99 point forty six percent, 99 point 62 percent, and 99 point 85 percent.

这个看起来GPU效果还是比CPU有极大提升的。

当然Mac Pro仅有Intel架构的机器，其中的机器及软件配置：

CPU数据跑的是TF2.3
GPU数据跑的是TF2.4 prerelease
机器配置：3.2GHz 16核 Intel Xeon W-based + 32GB内存 + AMD Radeon Pro Vega II Duo GPU （64GB HBM2显存） + 256GB SSD硬盘

看来想利用Intel架构的MacBook Pro来跑机器学习训练任务提升有限；用M1的MacBook Pro或Mac Pro可以跑机器学习训练任务了。

【来源参考】

Leveraging ML Compute for Accelerated Training on Mac https://machinelearning.apple.com/updates/ml-compute-training-on-mac

评论 6

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

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