想让人注意到某个有用的答案?请投赞同票!

有没有人帮助到了你?有没有任何答案或用户使用技巧解决了你的问题?选择“赞同票”箭头投出你的赞同票。你的反馈能帮助到他人!

进一步了解在什么情况下投赞同票:了解投票 - Apple 社区

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日 下午6:08

回复
回复量: 2

2024年5月1日 下午2: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 ID 开始畅游其中吧!