MLX vs MPS vs CUDA:苹果新机器学习框架的基准测试

发布时间 2023-12-21 10:10:58作者: deephub

如果你是一个Mac用户和一个深度学习爱好者,你可能希望在某些时候Mac可以处理一些重型模型。苹果刚刚发布了MLX,一个在苹果芯片上高效运行机器学习模型的框架。

最近在PyTorch 1.12中引入MPS后端已经是一个大胆的步骤,但随着MLX的宣布,苹果还想在开源深度学习方面有更大的发展。

在本文中,我们将对这些新方法进行测试,在三种不同的Apple Silicon芯片和两个支持cuda的gpu上和传统CPU后端进行基准测试。

 

https://avoid.overfit.cn/post/eb87d12f29eb4665adb43ad59fd3d64f