M2 pro芯片多核GPU支持

我在使用mac mini的高配记性,搭配了M2 pro的芯片,配置有16核GPU。通过安装metal支持TensorFlow的GPU插件/支持库,目前TensorFlow可以在mac上运行单核GPU神经网络训练加速。

但是我买的电脑配置有16核GPU,为了这个多核GPU数量,我还选择了更贵的配置版本。却发现自己多花钱买的pro芯片,跟普通mac mini版本相比并未获得实际的收益。

最近我查遍苹果的社区/网页/新闻和技术支持,都无法得到苹果是否在metal支持多核GPU的进展上有任何信息。

这样,我感到购买mac的高端多GPU核的设备,没有更多的实际效用。多CPU核并未有多少吸引力,AMD最新的Ryzen芯片更多CPU且运行ubuntu更容易获得应用更新和开发支持,此时还要计算从ubuntu的开发环境转到mac上的迁徙成本和学习成本。

能够获得在mac上支持多GPU核心的进展情况?如果确定以后不会支持多核GPU,我最终把生产环境升级到mac pro确实是全无动力。

Mac mini (M2, 2023)

发布日期 2024年4月28日 上午3:08

回复
回复量: 2

2024年4月30日 下午11:35 回应 msher_mac

先說結論:

當前的 TensorFlow-metal 不支持的是多 GPU 不是多核 GPU, 目前插件是把 GPU 當做一個整體來用的.

參考:

https://discuss.tensorflow.org/t/accelerating-tensorflow-using-apple-m1-max/5282/16

https://forums.developer.apple.com/forums/thread/690276


實驗代碼:

直接使用了Tensorflow Plugin - Metal - Apple Developer中的測試代碼


實驗環境:

Python 3.11.9

TensorFlow 2.16.1

TensorFlow-metal 1.1.0

M2 Max 64GB GPU 30


實驗還沒跑完, 跑完之後我會貼時間上來, 你可以對比一下.

这个主题已被系统或社区团队关闭。 你可以为你认为有帮助的任何帖子投票,也可以在社区中搜索其他答案。

M2 pro芯片多核GPU支持

欢迎来到 Apple 支持社区
Apple 客户在其产品方面互相帮助的论坛。使用您的 Apple 帐户开始畅游其中吧!!