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

2023-06-01 0 218

值班员爱家(ChinaZ.com)5月20日最新消息:据MacRumors报导,PyTorch与苹果公司的Metal工程建设项目组密切合作,近日正式宣布其开放源码机器学习架构将迅速全力支持在选用M1、M1 Pro、M1 Max或M1 Ultra晶片的Apple siliconMac上展开GPU快速数学模型体能训练。

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

相片源自 Apple

到为止,PyTorch在Mac上的体能训练只借助了CPU,但将要面世的版将容许开发相关人员和科学研究相关人员借助Apple silicon晶片中的软件系统GPU展开速率明显更慢的数学模型体能训练。

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

举报

相关文章

发表评论
暂无评论
官方客服团队

为您解决烦忧 - 24小时在线 专业服务