田海立@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:

結果顯示: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訓練的資料如下:

這個看起來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
CSDN認證博客專家
系統分析師
Android
人工智能
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/226618.html
標籤:AI
