{"id":204162,"date":"2025-05-29T09:47:39","date_gmt":"2025-05-29T01:47:39","guid":{"rendered":"https:\/\/server.hk\/cnblog\/204162\/"},"modified":"2025-05-29T09:47:39","modified_gmt":"2025-05-29T01:47:39","slug":"gpu%e8%ae%ad%e7%bb%83%e6%a8%a1%e5%9e%8b%e6%97%b6%ef%bc%8ccpu%e5%88%a9%e7%94%a8%e7%8e%87%e8%bf%87%e9%ab%98%e6%80%8e%e4%b9%88%e5%8a%9e%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/204162\/","title":{"rendered":"GPU\u8bad\u7ec3\u6a21\u578b\u65f6\uff0cCPU\u5229\u7528\u7387\u8fc7\u9ad8\u600e\u4e48\u529e\uff1f"},"content":{"rendered":"

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

GPU\u8bad\u7ec3\u6a21\u578b\u65f6\uff0cCPU\u5229\u7528\u7387\u8fc7\u9ad8\u600e\u4e48\u529e\uff1f<\/h1>\n<\/p>\n

<\/p>\n

\u95ee\u9898\u63cf\u8ff0<\/strong><\/p>\n

\u5728\u4f7f\u7528gpu\u8bad\u7ec3\u6a21\u578b\u65f6\uff0c\u4efb\u52a1\u7ba1\u7406\u5668\u663e\u793a\u663e\u5361\u5229\u7528\u7387\u8f83\u4f4e\uff0c\u800ccpu\u5229\u7528\u7387\u5374\u5f88\u9ad8\u3002 <\/p>\n

\u89e3\u51b3\u65b9\u6848<\/strong><\/p>\n

1. \u68c0\u67e5gpu\u914d\u7f6e<\/strong><\/p>\n

\u2460 \u6d4b\u8bd5cuda\u53ef\u7528\u6027<\/strong><\/p>\n

import torch\nprint(torch.cuda.is_available())<\/pre>\n
\u82e5\u8f93\u51fa\u4e3afalse\uff0c\u5219cuda\u672a\u6b63\u786e\u914d\u7f6e\u3002 <\/p>\n
\u2461 \u6307\u5b9agpu\u8bbe\u5907<\/strong><\/p>\n
import torch\ndevice = 'cuda:0' if torch.cuda.is_available() else 'cpu'\nfrom torchvision.models import ResNet\nResNet(......).to(device)<\/pre>\n
2. \u68c0\u67e5\u4ee3\u7801<\/strong><\/p>\n
\u2460 \u907f\u514d\u591a\u7ebf\u7a0b\/\u8fdb\u7a0b<\/strong><\/p>\n
\u5982\u679c\u4ee3\u7801\u4e2d\u5f00\u542f\u4e86\u591a\u7ebf\u7a0b\u6216\u591a\u8fdb\u7a0b\u8fdb\u884c\u6570\u636e\u5904\u7406\u6216\u753b\u56fe\uff0c\u5219cpu\u5229\u7528\u7387\u4f1a\u663e\u7740\u589e\u52a0\u3002\u5efa\u8bae\u51cf\u5c11\u8fd9\u4e9b\u64cd\u4f5c\u6216\u964d\u4f4e\u7ebf\u7a0b\/\u8fdb\u7a0b\u6570\u3002 <\/p>\n
\u2461 \u589e\u5927batch size<\/strong><\/p>\n
\u589e\u5927\u6279\u5904\u7406\u5927\u5c0f\u53ef\u4ee5\u63d0\u9ad8gpu\u5229\u7528\u7387\u3002\u8fd9\u5c06\u5360\u7528\u66f4\u591a\u663e\u5b58\uff0c\u51cf\u5c11\u56e0\u663e\u5b58\u4e0d\u8db3\u9020\u6210\u7684gpu\u95f2\u7f6e\u95ee\u9898\u3002 <\/p>\n
\u597d\u4e86\uff0c\u672c\u6587\u5230\u6b64\u7ed3\u675f\uff0c\u5e26\u5927\u5bb6\u4e86\u89e3\u4e86\u300aGPU\u8bad\u7ec3\u6a21\u578b\u65f6\uff0cCPU\u5229\u7528\u7387\u8fc7\u9ad8\u600e\u4e48\u529e\uff1f\u300b\uff0c\u5e0c\u671b\u672c\u6587\u5bf9\u4f60\u6709\u6240\u5e2e\u52a9\uff01\u5173\u6ce8\u516c\u4f17\u53f7\uff0c\u7ed9\u5927\u5bb6\u5206\u4eab\u66f4\u591a\u6587\u7ae0\u77e5\u8bc6\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"
GPU\u8bad\u7ec3\u6a21\u578b\u65f6\uff0cCPU\u5229\u7528\u7387\u8fc7...<\/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-204162","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204162","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=204162"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204162\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=204162"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=204162"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=204162"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}