{"id":431,"date":"2022-04-22T02:57:05","date_gmt":"2022-04-22T02:57:05","guid":{"rendered":"https:\/\/blog.liguanxin.cn\/?p=431"},"modified":"2022-06-30T02:59:42","modified_gmt":"2022-06-30T02:59:42","slug":"mindspore%e7%ae%97%e5%ad%90%e5%bc%80%e5%8f%91%e6%8c%87%e5%8d%97%ef%bc%881%ef%bc%89-gpu%e7%ae%97%e5%ad%90%e8%ae%be%e8%ae%a1%e6%96%87%e6%a1%a3","status":"publish","type":"post","link":"https:\/\/blog.liguanxin.cn\/index.php\/2022\/04\/22\/mindspore%e7%ae%97%e5%ad%90%e5%bc%80%e5%8f%91%e6%8c%87%e5%8d%97%ef%bc%881%ef%bc%89-gpu%e7%ae%97%e5%ad%90%e8%ae%be%e8%ae%a1%e6%96%87%e6%a1%a3\/","title":{"rendered":"MindSpore\u7b97\u5b50\u5f00\u53d1\u2014\u2014GPU\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863"},"content":{"rendered":"

\u9996\u5148\u4e0b\u8f7d\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863\u6a21\u677f\uff1aGPU\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863\u6a21\u677f\uff08\u70b9\u51fb\u4e0b\u8f7d\uff09<\/a><\/h3>\n
\n

\uff08\u4ee5\u4e0b\u5185\u5bb9\u4ee5\u7b97\u5b50FractionalAvgPool\u4e3a\u4f8b\uff09<\/p>\n

\u51c6\u5907\u90e8\u5206<\/h2>\n

\u9996\u5148\u901a\u8fc7\u641c\u7d22\u5f15\u64ce\u67e5\u4e00\u4e0b\u76f8\u5e94\u7684\u7b97\u5b50\u6709\u4ec0\u4e48\u4f5c\u7528\uff1a
\n\"\"
\n\u5927\u6982\u4e86\u89e3\u8fd9\u4e2a\u7b97\u5b50\u662f\u505a\u4e0d\u89c4\u5219\u533a\u57df\u7684\u5e73\u5747\u6c60\u5316\u7684\u3002
\n\u5bf9\u4e8e\u9700\u8981\u5f00\u53d1\u7684\u7b97\u5b50\uff0c\u9996\u5148\u8fdb\u5165MindSpore\u7684issues\u680f\u641c\u7d22\uff1ahttps:\/\/toscode.gitee.com\/mindspore\/mindspore\/issues<\/a>
\n\u56e0\u4e3a\u6211\u9700\u8981\u5f00\u53d1\u7684\u662fFractionalAvgPool\uff0c\u6240\u4ee5\u6211\u4f1a\u641c\u7d22\u5230\u8fd9\u6837\u7684\u7b97\u5b50\u4fe1\u606f\uff1a
\n\"\"
\n\"\"
\n\"\"
\n\u6839\u636e\u6b64\u5904\u7684\u6807\u6746\u63a5\u53e3\u53c2\u8003\uff0c\u53ef\u4ee5\u770b\u5230\u5728tensorflow\u4e2d\u8fd9\u4e2a\u7b97\u5b50\u7684\u5b9e\u73b0\uff1a
\n\"\"
\n\u7136\u540e\u5c31\u53ef\u4ee5\u5f00\u59cb\u5199\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863\u4e86\u3002<\/p>\n

\u529f\u80fd\u4ecb\u7ecd\u548c\u63a5\u53e3\u63cf\u8ff0<\/h2>\n

\u8fd9\u90e8\u5206\u53ef\u4ee5\u76f4\u63a5\u53c2\u8003MindSpore\u7684issues\u90a3\u90e8\u5206\u586b\u5165\u3002<\/p>\n

\u5b9e\u73b0\u65b9\u5f0f\uff08\u91cd\u70b9\uff09<\/h2>\n

\u5b9e\u73b0\u65b9\u5f0f\u4e3b\u8981\u662f\u53c2\u8003pytorch\u6216\u8005tensorflow\u7684\u6e90\u7801\u3002<\/p>\n

\u901a\u8fc7git\u4e0b\u8f7dtensorflow 2.6.2\u7248\u4ee3\u7801<\/p>\n

git clone --branch v2.6.2 https:\/\/github.com\/tensorflow\/tensorflow.git<\/code><\/pre>\n
\u5982\u679c\u662f\u4e0b\u8f7dpytorch 1.8.1<\/p>\n
git clone --branch v1.8.1 https:\/\/github.com\/pytorch\/pytorch.git<\/code><\/pre>\n
\u5982\u679c\u4e0b\u8f7d\u592a\u6162\u53ef\u4f7f\u7528
\nhttps:\/\/github.com\/tensorflow\/tensorflow\/archive\/refs\/tags\/v2.6.2.zip<\/a>
\n\u548c
\nhttps:\/\/github.com\/pytorch\/pytorch\/archive\/refs\/tags\/v1.8.1.zip<\/a><\/p>\n
tensorflow<\/strong>
\n\u901a\u8fc7vscode\u6216pycharm\u641c\u7d22tensorflow\u4e2d\u7684\u63a5\u53e3\u5b9a\u4e49\uff0c\u76f4\u63a5\u641c\u7d22REGISTER_OP("\u7b97\u5b50\u540d\u79f0")\uff0c\u5176\u4e2d\u7b97\u5b50\u540d\u79f0\u7528\u5927\u9a7c\u5cf0\u547d\u540d\u3002
\n\u4e8e\u662fFractionalAvgPool\u8fd9\u4e2a\u7b97\u5b50\u53ef\u4ee5\u5b9a\u4f4d\u5230\u76f8\u5173\u63a5\u53e3\u5b9a\u4e49\u4e3a\uff1a
\n\"\"
\n\u4e0e\u524d\u9762\u7684\u53c2\u6570\u5b9a\u4e49\u57fa\u672c\u4e00\u81f4\u3002<\/p>\n
\u7136\u540e\u518d\u627e\u76f8\u5e94\u7684cuda\u4ee3\u7801\uff0c\u5927\u90e8\u5206\u7b97\u5b50\u5728\u76ee\u5f55\\tensorflow\\tensorflow\\core\\kernels\u4e0b\uff0ccuda\u5b9e\u73b0\u7684\u7b97\u5b50\u540e\u7f00\u4e3acu.cc\uff0c\u5934\u6587\u4ef6\u540e\u7f00\u4e3a.h\uff0c\u5176\u4ed6\u90fd\u662fCPU\u5b9e\u73b0\u3002\u627e\u5230\u76f8\u5e94\u7b97\u5b50\u7684GPU\u5b9e\u73b0\uff0c\u7136\u540e\u628aGPU\u5b9e\u73b0\u8bfb\u61c2\uff0c\u7136\u540e\u628a\u5b9e\u73b0\u903b\u8f91\u5199\u5165\u8bbe\u8ba1\u6587\u6863\u4e2d\u3002\uff08\u9047\u5230\u4e0d\u61c2\u7684\u53ef\u4ee5\u770b\u8fd9\u7bc7\u535a\u5ba2<\/strong><\/a>\u7684cuda\u7f16\u7a0b\u5165\u95e8\uff09<\/p>\n
pytorch<\/strong>
\n\u67e5\u627e\u7b97\u5b50\u5728pytorch\u4e2d\u7684cuda\u5b9e\u73b0\uff0c\u901a\u5e38\u8def\u5f84\u4e3a\\pytorch-1.8.1\\aten\\src\\ATen\\native\\cuda\uff0c\u540e\u7f00\u4e3a.cu<\/p>\n
\u7136\u540e\u8bfb\u61c2\u8fd9\u4e9bcuda\u5b9e\u73b0\uff0c\u628a\u81ea\u5df1\u603b\u7ed3\u7684\u5b9e\u73b0\u65b9\u5f0f\u5199\u5165\u6587\u6863\u3002<\/h2>\n","protected":false},"excerpt":{"rendered":"
\u9996\u5148\u4e0b\u8f7d\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863\u6a21\u677f\uff1aGPU\u7b97\u5b50\u8bbe\u8ba1\u6587\u6863\u6a21\u677f\uff08\u70b9\u51fb\u4e0b\u8f7d\uff09 \uff08\u4ee5\u4e0b\u5185\u5bb9\u4ee5\u7b97\u5b50FractionalAvgPoo […]<\/p>\n","protected":false},"author":1,"featured_media":215,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[29],"tags":[30,31,32],"_links":{"self":[{"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/posts\/431"}],"collection":[{"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/comments?post=431"}],"version-history":[{"count":0,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/posts\/431\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/media\/215"}],"wp:attachment":[{"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/media?parent=431"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/categories?post=431"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/blog.liguanxin.cn\/index.php\/wp-json\/wp\/v2\/tags?post=431"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}