針對Mac優化後的TensorFlow開始施展新的M1與其GPU的千里
機器學習環境TensorFlow的一個新的Mac優化分支發布新版,帶來重大的性能提升,在這之前GPU還沒有用於訓練任務。現在,基於M1處理器的設備已經看到了利用率被極大提升,這表明像這樣的一連串工作流優化即將到來。
TensorFlow在其博客上公佈,改進後的Mac版本在最好的情況下對於常見的訓練任務,速度提升了10倍以上。
對於任何從事ML工作的人來說,這本身就值得慶祝,但事實上,之前的TF版本也只利用了Mac上的CPU,而沒有利用GPU中強大的並行處理器,大多數大規模的ML訓練都是利用雲計算來完成的。
從只用CPU到CPU+GPU的變化可能是很大的改進原因,因為在基於英特爾的Mac Pro上的基準測試顯示在相同的硬件上有巨大的收益,曾經在6-8秒範圍內的訓練時間現在以秒的分數來衡量。