{"id":205019,"date":"2025-05-29T13:50:04","date_gmt":"2025-05-29T05:50:04","guid":{"rendered":"https:\/\/server.hk\/cnblog\/205019\/"},"modified":"2025-05-29T13:50:04","modified_gmt":"2025-05-29T05:50:04","slug":"%e5%a6%82%e4%bd%95%e5%9c%a8-pytorch-%e4%b8%ad%e5%88%a9%e7%94%a8-apple-silicon-%e7%9a%84%e7%a5%9e%e7%bb%8f%e7%bd%91%e7%bb%9c%e5%bc%95%e6%93%8e-npu%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/205019\/","title":{"rendered":"\u5982\u4f55\u5728 PyTorch \u4e2d\u5229\u7528 Apple Silicon \u7684\u795e\u7ecf\u7f51\u7edc\u5f15\u64ce (NPU)\uff1f"},"content":{"rendered":"

<\/b> <\/p>\n

\u5982\u4f55\u5728 PyTorch \u4e2d\u5229\u7528 Apple Silicon \u7684\u795e\u7ecf\u7f51\u7edc\u5f15\u64ce (NPU)\uff1f<\/h1>\n

\u5927\u5bb6\u597d\uff0c\u6211\u4eec\u53c8\u89c1\u9762\u4e86\u554a~\u672c\u6587\u7684\u5185\u5bb9\u4e2d\u5c06\u4f1a\u6d89\u53ca\u5230<\/span>\u7b49\u7b49\u3002\u5982\u679c\u4f60\u6b63\u5728\u5b66\u4e60\u6587\u7ae0<\/span>\u76f8\u5173\u77e5\u8bc6\uff0c\u6b22\u8fce\u5173\u6ce8\u6211\uff0c\u4ee5\u540e\u4f1a\u7ed9\u5927\u5bb6\u5e26\u6765\u66f4\u591a\u6587\u7ae0<\/span>\u76f8\u5173\u6587\u7ae0\uff0c\u5e0c\u671b\u6211\u4eec\u80fd\u4e00\u8d77\u8fdb\u6b65\uff01\u4e0b\u9762\u5c31\u5f00\u59cb\u672c\u6587\u7684\u6b63\u5f0f\u5185\u5bb9~<\/p>\n

<\/p>\n

\u5982\u4f55\u5728 pytorch \u4e2d\u5229\u7528 apple silicon \u7684\u795e\u7ecf\u7f51\u7edc\u5f15\u64ce (npu)<\/strong><\/p>\n

apple silicon \u62e5\u6709 cpu\u3001gpu \u548c npu \u8fd9 3 \u79cd\u5904\u7406\u5668\u3002\u6839\u636e\u5b98\u65b9\u6587\u6863\uff0cpytorch \u4e2d\u7684 mps \u540e\u7aef\u53ef\u4ee5\u5229\u7528 apple silicon \u4e2d\u7684 gpu\uff0c\u4f46\u65e0\u6cd5\u76f4\u63a5\u8bbf\u95ee npu\u3002<\/p>\n

\u4e3a\u4ec0\u4e48 mps \u65e0\u6cd5\u8bbf\u95ee npu<\/strong><\/p>\n

npu \u901a\u5e38\u7528\u4e8e\u63a8\u7406\u4efb\u52a1\uff0c\u800c\u4e0d\u662f\u8bad\u7ec3\u4efb\u52a1\u3002\u63a8\u7406\u662f\u6307\u4f7f\u7528\u8bad\u7ec3\u597d\u7684\u6a21\u578b\u5bf9\u65b0\u6570\u636e\u8fdb\u884c\u9884\u6d4b\u3002\u800c\u8bad\u7ec3\u662f\u6307\u521b\u5efa\u548c\u5fae\u8c03\u6a21\u578b\u4ee5\u63d0\u9ad8\u5176\u51c6\u786e\u6027\u3002<\/p>\n

\u5982\u4f55\u4f7f\u7528 mps<\/strong><\/p>\n

\u8981\u4f7f\u7528 mps \u540e\u7aef\uff0c\u53ef\u4ee5\u5728 pytorch \u4e2d\u5c06\u8bbe\u5907\u6307\u5b9a\u4e3a “mps”\uff1a<\/p>\n

import torch\ndevice = torch.device('mps')<\/pre>\n
\u4f7f\u7528 mps \u540e\u7aef\u53ef\u4ee5\u52a0\u901f pytorch \u64cd\u4f5c\uff0c\u56e0\u4e3a\u5b83\u76f4\u63a5\u8bbf\u95ee apple silicon gpu \u7684\u786c\u4ef6\u7279\u6027\u3002<\/p>\n
\u5982\u4f55\u4f7f\u7528 npu<\/strong><\/p>\n
\u82f9\u679c\u76ee\u524d\u5c1a\u672a\u5f00\u653e npu \u7f16\u7a0b\u3002\u5982\u679c\u4f60\u9700\u8981\u5728 apple silicon \u4e0a\u8fdb\u884c\u63a8\u7406\u4efb\u52a1\uff0c\u53ef\u4ee5\u4f7f\u7528 coreml \u6846\u67b6\uff0c\u5b83\u53ef\u4ee5\u81ea\u52a8\u5229\u7528 npu\u3002<\/p>\n
\u603b\u7ed3<\/strong><\/p>\n
    \n
  • mps \u540e\u7aef\u7528\u4e8e\u8bbf\u95ee apple silicon gpu\uff0c\u800c\u4e0d\u662f npu\u3002<\/li>\n
  • npu \u7528\u4e8e\u63a8\u7406\u4efb\u52a1\uff0c\u76ee\u524d\u4e0d\u80fd\u76f4\u63a5\u901a\u8fc7 pytorch \u7f16\u7a0b\u3002<\/li>\n
  • \u8981\u5728 apple silicon \u4e0a\u4f7f\u7528 npu\uff0c\u53ef\u4ee5\u4f7f\u7528 coreml \u6846\u67b6\u3002<\/li>\n<\/ul>\n
    \u672c\u7bc7\u5173\u4e8e\u300a\u5982\u4f55\u5728 PyTorch \u4e2d\u5229\u7528 Apple Silicon \u7684\u795e\u7ecf\u7f51\u7edc\u5f15\u64ce (NPU)\uff1f\u300b\u7684\u4ecb\u7ecd\u5c31\u5230\u6b64\u7ed3\u675f\u5566\uff0c\u4f46\u662f\u5b66\u65e0\u6b62\u5883\uff0c\u60f3\u8981\u4e86\u89e3\u5b66\u4e60\u66f4\u591a\u5173\u4e8e\u6587\u7ae0\u7684\u76f8\u5173\u77e5\u8bc6\uff0c\u8bf7\u5173\u6ce8\u516c\u4f17\u53f7\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"
    \u5982\u4f55\u5728 PyTorch \u4e2d\u5229\u7528 ...<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4925],"tags":[],"class_list":["post-205019","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/205019","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=205019"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/205019\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=205019"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=205019"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=205019"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}