机器学习框架PyTorch在苹果M1系列Mac上实现GPU加速训练

站长之家（ChinaZ.com）5月20日消息:据MacRumors报道，PyTorch与苹果公司的Metal工程团队合作，日前宣布其开源机器学习框架将很快支持在采用M1、M1 Pro、M1 Max或M1 Ultra芯片的Apple siliconMac上进行GPU加速模型训练。

到目前为止，PyTorch在Mac上的训练只利用了CPU，但即将推出的版本将允许开发者和研究人员利用Apple silicon芯片中的集成GPU进行速度显著更快的模型训练。

带有GPU加速训练的PyTorch1.12版预览版可用于运行macOS12.3或更高版本的Apple silicon Mac，并带有Python的本地版本。性能比较和其他细节可在PyTorch的网站上找到。

上一条：为何 NVidia 的开源驱动程序对 Linux 如此重要？

下一条：厌倦了Windows却买不起Mac？请查收这份Linux版本选择指南