{"id":14802,"date":"2023-03-14T08:54:51","date_gmt":"2023-03-14T00:54:51","guid":{"rendered":"http:\/\/139.9.1.231\/?p=14802"},"modified":"2023-03-14T08:54:52","modified_gmt":"2023-03-14T00:54:52","slug":"torch-cuda-empty_cache","status":"publish","type":"post","link":"http:\/\/139.9.1.231\/index.php\/2023\/03\/14\/torch-cuda-empty_cache\/","title":{"rendered":"torch.cuda.empty_cache \u6e05\u7a7a\u663e\u5b58"},"content":{"rendered":"\n<figure class=\"wp-block-image size-large\"><img loading=\"lazy\" width=\"1024\" height=\"412\" src=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2023\/03\/image-15-1024x412.png\" alt=\"\" class=\"wp-image-14805\" srcset=\"http:\/\/139.9.1.231\/wp-content\/uploads\/2023\/03\/image-15-1024x412.png 1024w, http:\/\/139.9.1.231\/wp-content\/uploads\/2023\/03\/image-15-300x121.png 300w, http:\/\/139.9.1.231\/wp-content\/uploads\/2023\/03\/image-15-768x309.png 768w, http:\/\/139.9.1.231\/wp-content\/uploads\/2023\/03\/image-15.png 1335w\" sizes=\"(max-width: 1024px) 100vw, 1024px\" \/><\/figure>\n\n\n\n<p>       empty_cache\uff08\uff09\u4e0d\u4f1a\u589e\u52a0Pytorch\u53ef\u7528\u7684GPU\u5185\u5b58\u91cf\u3002\u4f46\u662f\uff0c\u5728\u67d0\u4e9b\u60c5\u51b5\u4e0b\uff0c\u5b83\u53ef\u80fd\u6709\u52a9\u4e8e\u51cf\u5c11GPU\u8bb0\u5fc6\u7684\u788e\u7247\u5316\u3002\u6709\u5173GPU\u5185\u5b58\u7ba1\u7406\u7684\u66f4\u591a\u8be6\u7ec6\u4fe1\u606f\uff0c\u8bf7\u53c2\u89c1\u5185\u5b58\u7ba1\u7406\u3002<\/p>\n\n\n\n<p>       \u56e0\u4e3aPyTorch\u662f\u6709\u7f13\u5b58\u533a\u7684\u8bbe\u7f6e\u7684\uff0c\u610f\u601d\u5c31\u662f\u4e00\u4e2aTensor\u5c31\u7b97\u88ab\u91ca\u653e\u4e86\uff0c\u8fdb\u7a0b\u4e5f\u4e0d\u4f1a\u628a\u7a7a\u95f2\u51fa\u6765\u7684\u663e\u5b58\u8fd8\u7ed9GPU\uff0c\u800c\u662f\u7b49\u5f85\u4e0b\u4e00\u4e2aTensor\u6765\u586b\u5165\u8fd9\u4e00\u7247\u88ab\u91ca\u653e\u7684\u7a7a\u95f4\u3002\u6240\u4ee5\u6211\u4eec\u7528nvidia-smi\/gpustat \u770b\u5230\u7684\u663e\u5b58\u5360\u7528\u4e0d\u4f1a\u51cf\u5c11<\/p>\n\n\n\n<p class=\"has-light-gray-background-color has-background\">\u7528torch.cuda.empty_cache\u53ef\u4ee5\u6e05\u7a7a\u7f13\u51b2\u533a\uff1a<\/p>\n\n\n\n<p>\u5728\u7a0b\u5e8f\u4e2d\u52a0\u4e0a\u8fd9\u53e5\u4f1a\u4f7f\u901f\u5ea6\u53d8\u6162\u4e00\u4e9b\uff0c\u4f46\u662f\u6709\u4e9b\u60c5\u51b5\u4e0b\u4f1a\u6709\u7528\uff0c\u4f8b\u5982\u7a0b\u5e8f\u4e4b\u524dtest\u7684\u65f6\u5019\u603b\u662f\u7206\u663e\u5b58\uff0c\u7136\u540e\u5728\u5faa\u73af\u4e2d\u52a0\u4e0a\u4e86\u8fd9\u53e5\u5c31\u4e0d\u7206\u4e86<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code>for i, data in enumerate(data_loader):\n    torch.cuda.empty_cache()\n    img_meta = data&#91;'img_meta']&#91;0].data&#91;0]\n    img_name = img_meta&#91;0]&#91;'filename'].split('\/')&#91;-1]\n    with torch.no_grad():\n        result = model(return_loss=False, rescale=not show, **data)<\/code><\/pre>\n\n\n\n<p>\u5982\u679c\u663e\u5b58\u8d44\u6e90\u6bd4\u8f83\u7d27\u7f3a\uff0c\u53ef\u4ee5\u5728\u6bcf\u4e2aepoch\u5f00\u59cb\u65f6\u91ca\u653e\u4e0b\u4e0d\u7528\u7684\u663e\u5b58\u8d44\u6e90\u3002<\/p>\n\n\n\n<p>torch.cuda.empty_cache() # \u91ca\u653e\u663e\u5b58<br>\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014\u2014<\/p>\n","protected":false},"excerpt":{"rendered":"<p>empty_cache\uff08\uff09\u4e0d\u4f1a\u589e\u52a0Pytorch\u53ef\u7528\u7684GPU\u5185\u5b58\u91cf\u3002\u4f46\u662f\uff0c\u5728\u67d0\u4e9b\u60c5\u51b5\u4e0b\uff0c\u5b83\u53ef\u80fd\u6709\u52a9\u4e8e\u51cf\u5c11GPU &hellip; <a href=\"http:\/\/139.9.1.231\/index.php\/2023\/03\/14\/torch-cuda-empty_cache\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">torch.cuda.empty_cache \u6e05\u7a7a\u663e\u5b58<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[11],"tags":[],"_links":{"self":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/14802"}],"collection":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/comments?post=14802"}],"version-history":[{"count":5,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/14802\/revisions"}],"predecessor-version":[{"id":14809,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/14802\/revisions\/14809"}],"wp:attachment":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/media?parent=14802"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/categories?post=14802"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/tags?post=14802"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}