首页 > 最新动态 > 苹果M1 Mac:支持PyTorch训练使用GPU加速

苹果M1 Mac:支持PyTorch训练使用GPU加速

时间:2022-05-20 | 作者:老毛桃

大家应该都知道,Pytorch可以在Mac上仅支持使用CPU进行训练。而最近,Pytorch官方公布了新的消息,表示最新版v1.12可以支持GPU加速了,前提是搭载M1系列芯片的Mac产品。

其实,这个新功能是Pytorch和Apple Metal工程团队合作推出的,简单来说,就是通过Apple MPS作为PyTorch后端启用GPU加速训练。同时,为了进一步优化计算性能,MPS还针对Metal GPU进行了特定的微调。

苹果M1 Mac:支持PyTorch训练使用GPU加速

这里面的Metal,其实是一个和OpenGL类似的框架,但是各自的适用范围不太一样。其中,OpenGL适用于各平台移动端GPU的渲染计算,而Metal则是专属于iOS / MacOS平台的渲染计算,它可以直接调用GPU进行图形处理等工作。

关于本次的新功能,苹果也进行了很多的测试,就测试结果可知,相比CPU加速,通过GPU加速可以将模型训练速度提高约7倍,评估速度也能提高约20倍。其实,它的整体性能对比Nvidia GPU 的laptop来说还不是很够,但是呢,它在功耗方面还不错,所以Mac Studio有可能会真香哦。

毕竟,就现有的设备来说,它是你能找到的4800 美元价位内,最具性价比的机器了。如果你想使用这个功能的话,需要注意的是,你的苹果macOS操作系统版本需要在12.3版本或以上。

同时,电脑还需要安装arm64原生Python,再去官网下载最新版Pytorch预览版就可以了。如果你想了解更多的信息,可以去官网进行查看。