首页
/ Segment Anything 2项目新增对MPS和CPU设备的支持

Segment Anything 2项目新增对MPS和CPU设备的支持

2025-05-15 05:55:24作者:裴锟轩Denise

Segment Anything 2(SAM2)作为Facebook Research推出的先进图像分割模型,近期迎来了一个重要更新——正式支持苹果M系列芯片的Metal Performance Shaders(MPS)以及CPU设备。这一更新解决了Mac用户和没有NVIDIA GPU设备用户的使用难题。

技术背景

在计算机视觉领域,图像分割模型通常需要强大的GPU加速计算。传统上,这类模型主要依赖NVIDIA的CUDA架构进行加速。然而,随着苹果自研芯片的普及,越来越多的开发者希望在Mac设备上直接运行这些模型。

MPS是苹果提供的Metal框架中的高性能计算组件,专门为苹果芯片优化。它能够充分利用M系列芯片的统一内存架构和强大的神经网络引擎,为深度学习推理提供接近CUDA的性能。

更新内容

此次更新主要涉及以下方面:

  1. 设备兼容性扩展:模型现在可以自动检测并适配MPS设备(Mac)、CUDA设备(NVIDIA GPU)或CPU设备
  2. 性能优化:针对MPS设备进行了专门的优化,确保在苹果芯片上获得最佳性能
  3. 示例更新:项目中的Jupyter Notebook示例已更新,支持在不同设备上运行

使用指南

对于Mac用户,现在可以无需任何额外配置直接运行SAM2模型。系统会自动检测并使用MPS加速。对于没有GPU设备的用户,模型也可以回退到CPU模式运行,虽然速度会有所下降,但保证了功能的可用性。

技术意义

这一更新具有多重意义:

  • 降低了使用门槛:使更多没有高端NVIDIA GPU设备的开发者能够体验SAM2的强大功能
  • 生态扩展:完善了苹果生态下的深度学习工具链
  • 研究便利性:研究人员可以在更多设备上进行实验和原型开发

未来展望

随着异构计算的发展,支持多种计算后端将成为深度学习框架的标配。SAM2的这一更新也反映了这一趋势,为后续更多设备支持奠定了基础。

对于想要在Mac设备上体验先进图像分割技术的开发者来说,这无疑是一个令人振奋的消息。现在,M系列Mac用户可以直接本地运行SAM2,无需依赖云服务或其他硬件设备。

登录后查看全文
热门项目推荐