首页>软件资讯>行业资讯

行业资讯

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

发布时间:2022-06-13 08:53:04人气:345

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


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

image.png

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



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

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