站長(zhǎng)之家(ChinaZ.com)5月20日消息:據(jù)MacRumors報(bào)道,PyTorch與蘋果公司的Metal工程團(tuán)隊(duì)合作,日前宣布其開源機(jī)器學(xué)習(xí)框架將很快支持在采用M1、M1 Pro、M1 Max或M1 Ultra芯片的Apple siliconMac上進(jìn)行GPU加速模型訓(xùn)練。
圖片來(lái)自 Apple
到目前為止,PyTorch在Mac上的訓(xùn)練只利用了CPU,但即將推出的版本將允許開發(fā)者和研究人員利用Apple silicon芯片中的集成GPU進(jìn)行速度顯著更快的模型訓(xùn)練。
帶有GPU加速訓(xùn)練的PyTorch1.12版預(yù)覽版可用于運(yùn)行macOS12.3或更高版本的Apple silicon Mac,并帶有Python的本地版本。性能比較和其他細(xì)節(jié)可在PyTorch的網(wǎng)站上找到。
(舉報(bào))