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)?的详细内容,更多请关注【创想鸟】其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至253000106@qq.com举报,一经查实,本站将立刻删除。
发布者:PHP中文网,转转请注明出处:https://www.chuangxiangniao.com/p/2181984.html