PyTorch 如何使用 Apple Silicon 神经网络引擎 (NPU)？

从现在开始，努力学习吧！本文《PyTorch 如何使用 Apple Silicon 神经网络引擎 (NPU)？》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

PyTorch 如何利用 Apple Silicon 神经网络引擎 (NPU)

苹果 M1 系列芯片搭载了 CPU、GPU 和神经网络引擎 (NPU)。PyTorch 可让您指定设备，如 MPS（金属性能阴影），该设备可在 Apple Silicon 上运行。

MPS 调用的是 GPU 还是 NPU？

MPS 仅调用 GPU，而不是 NPU。GPU 用于训练和生成向量。

如何选择 NPU？

目前，NPU 仅用于推理，而非训练。无法在 PyTorch 中直接选择 NPU。要使用 NPU，需要使用 CoreML，并且苹果尚未公开 NPU 编程接口。

到这里，我们也就讲完了《PyTorch 如何使用 Apple Silicon 神经网络引擎 (NPU)？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！