{"id":658,"date":"2025-01-03T09:44:04","date_gmt":"2025-01-03T01:44:04","guid":{"rendered":"https:\/\/wangliguang.cn\/?p=658"},"modified":"2025-01-03T09:44:04","modified_gmt":"2025-01-03T01:44:04","slug":"%e3%80%90%e6%96%b0%e6%95%99%e7%a8%8b%e3%80%91%e5%8d%8e%e4%b8%ba%e6%98%87%e8%85%benpu%e7%9a%84pytorch%e7%8e%af%e5%a2%83%e6%90%ad%e5%bb%ba","status":"publish","type":"post","link":"https:\/\/wangliguang.cn\/?p=658","title":{"rendered":"\u3010\u65b0\u6559\u7a0b\u3011\u534e\u4e3a\u6607\u817eNPU\u7684pytorch\u73af\u5883\u642d\u5efa"},"content":{"rendered":"<h2>1 \u786c\u4ef6\u914d\u7f6e<\/h2>\n<p>\u4f7f\u7528\u5b66\u6821\u7684\u96c6\u7fa4\uff0c\u76f8\u5173\u914d\u7f6e\u5982\u4e0b\uff1a<br \/>\nCPU\uff1a\u9cb2\u9e4f920<br \/>\nNPU\uff1a\u6607\u817e910B<br \/>\n\u64cd\u4f5c\u7cfb\u7edf\uff1aopenEuler 22.03<\/p>\n<h2>2 \u5b89\u88c5\u7248\u672c<\/h2>\n<p>\u6839\u636e\u6607\u817e\u5b98\u65b9gitee\u4e0a\u7684\u4fe1\u606f\uff0cPytoch 2.1.0\u662f\u957f\u671f\u652f\u6301\u7248\u672c\uff0c\u56e0\u6b64\u9009\u62e9\u5b89\u88c5\u8fd9\u4e00\u7248\u672c\uff0c\u4ece\u800c\u6700\u5927\u9650\u5ea6\u907f\u5751\u3002<br \/>\n\u672c\u6559\u7a0b\u9009\u62e9\u7684\u7248\u672c\u5982\u4e0b\uff1a<br \/>\nPython\uff1a3.9<br \/>\nPytorch\uff1a2.1.0<br \/>\n\u9009\u62e9\u4f9d\u636e\uff1a<br \/>\n<img decoding=\"async\" src=\"https:\/\/liguang.wang\/usr\/uploads\/2025\/01\/916116704.png\" alt=\"1.PNG\" \/><br \/>\n<img decoding=\"async\" src=\"https:\/\/liguang.wang\/usr\/uploads\/2025\/01\/2066984537.png\" alt=\"2.PNG\" \/><br \/>\n\u53c2\u8003\u94fe\u63a5\uff1a<a href=\"https:\/\/gitee.com\/ascend\/pytorch\">https:\/\/gitee.com\/ascend\/pytorch<\/a><\/p>\n<h2>3 \u5b89\u88c5\u6b65\u9aa4<\/h2>\n<p>1\u3001\u65b0\u5efaconda\u73af\u5883\u3002<\/p>\n<pre><code>conda create -n pytorch python=3.9<\/code><\/pre>\n<p>2\u3001\u5728\u65b0\u5efa\u597d\u7684conda\u73af\u5883\u4e2d\uff0c\u5b89\u88c5\u57fa\u7840\u7684\u4f9d\u8d56\u3002<\/p>\n<pre><code>pip install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py --user<\/code><\/pre>\n<p>\u53c2\u8003\u94fe\u63a5\uff1a<br \/>\n<a href=\"https:\/\/www.hiascend.com\/document\/detail\/zh\/canncommercial\/80RC3\/softwareinst\/instg\/instg_0006.html?Mode=PmIns&amp;OS=Ubuntu&amp;Software=cannToolKit\">https:\/\/www.hiascend.com\/document\/detail\/zh\/canncommercial\/80RC3\/softwareinst\/instg\/instg_0006.html?Mode=PmIns&OS=Ubuntu&Software=cannToolKit<\/a><br \/>\n3\u3001\u5b89\u88c5Pytoch<\/p>\n<pre><code>wget https:\/\/download.pytorch.org\/whl\/cpu\/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=de7d63c6ecece118684415a3dbd4805af4a4c1ee1490cccf7405d8c240a481b4<\/code><\/pre>\n<p>\u5728\u4e0a\u9762\u5efa\u597d\u7684\u73af\u5883\u4e2dpip install\u4e0b\u8f7d\u597d\u7684\u5305\u3002<br \/>\n4\u3001\u5b89\u88c5pytorch-npu<br \/>\n\u65b9\u6cd5\u4e00\uff1a\u624b\u52a8\u67e5\u627e\u5bf9\u5e94\u5173\u7cfb\u3002<br \/>\n\u9996\u5148\u786e\u5b9a\u81ea\u5df1\u7684CANN\u7248\u672c\u3002\u5982\u679c\u4e0d\u77e5\u9053\u7528\u7684\u662f\u54ea\u4e00\u4e2aCANN\uff0c\u53ef\u4ee5\u5148Python\u4e00\u4e0b\uff0cimport torch\u8fd9\u4e2a\u5305\uff0c\u7136\u540e\u770b\u544a\u8b66\u4fe1\u606f\u662f\u54ea\u4e2a\u7248\u672c\u7684\u3002<br \/>\n\u786e\u5b9a\u597dCANN\u7248\u672c\u540e\uff0c\u4ece\u4e0b\u8868\u4e2d\u624b\u52a8\u67e5\u627e\u5bf9\u5e94\u7684torch_npu\u7248\u672c\u3002<br \/>\n<img decoding=\"async\" src=\"https:\/\/liguang.wang\/usr\/uploads\/2025\/01\/3524635029.png\" alt=\"3.PNG\" \/><br \/>\n\u65b9\u6cd5\u4e8c\uff1a\u81ea\u52a8\u5339\u914d\u3002<br \/>\n\u4ece\u4ee5\u4e0b\u7f51\u7ad9\uff0c\u6839\u636e\u5339\u914d\u7ed3\u679c\u4e0b\u8f7d\uff1a<a href=\"https:\/\/www.hiascend.com\/developer\/download\/community\/result?module=pt+cann&amp;pt=6.0.RC3.beta1&amp;cann=8.0.RC3.beta1\">https:\/\/www.hiascend.com\/developer\/download\/community\/result?module=pt+cann&pt=6.0.RC3.beta1&cann=8.0.RC3.beta1<\/a><br \/>\n\u4e0b\u8f7d\u597d\u6307\u5b9a\u7248\u672c\u540e\uff0c\u4f7f\u7528pip install\u5b89\u88c5\u4e0b\u8f7d\u597d\u7684\u5305\u5373\u53ef<br \/>\n5\u3001\u6d4b\u8bd5\u3002<br \/>\n\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u6d4b\u8bd5\uff0c\u5982\u679c\u6b63\u5e38\u8f93\u51fa\u7ed3\u679c\uff0c\u8bf4\u660e\u73af\u5883\u642d\u5efa\u6210\u529f\u3002<\/p>\n<pre><code>python3 -c &quot;import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);&quot;<\/code><\/pre>\n<p>\u7ed3\u679c\u5982\u4e0b\uff1a<br \/>\n<img decoding=\"async\" src=\"https:\/\/liguang.wang\/usr\/uploads\/2025\/01\/2199440669.png\" alt=\"4.PNG\" \/><\/p>\n","protected":false},"excerpt":{"rendered":"<p>1 \u786c\u4ef6\u914d\u7f6e \u4f7f\u7528\u5b66\u6821\u7684\u96c6\u7fa4\uff0c\u76f8\u5173\u914d\u7f6e\u5982\u4e0b\uff1a CPU\uff1a\u9cb2\u9e4f920 NPU\uff1a\u6607\u817e910B \u64cd\u4f5c\u7cfb\u7edf\uff1aopenEu&hellip; <a href=\"https:\/\/wangliguang.cn\/?p=658\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb <span class=\"screen-reader-text\">\u3010\u65b0\u6559\u7a0b\u3011\u534e\u4e3a\u6607\u817eNPU\u7684pytorch\u73af\u5883\u642d\u5efa<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4,8],"tags":[],"class_list":["post-658","post","type-post","status-publish","format-standard","hentry","category-diy","category-linux"],"_links":{"self":[{"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/posts\/658","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=658"}],"version-history":[{"count":1,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/posts\/658\/revisions"}],"predecessor-version":[{"id":659,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=\/wp\/v2\/posts\/658\/revisions\/659"}],"wp:attachment":[{"href":"https:\/\/wangliguang.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=658"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=658"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/wangliguang.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=658"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}