{"id":204757,"date":"2025-05-29T16:29:45","date_gmt":"2025-05-29T08:29:45","guid":{"rendered":"https:\/\/server.hk\/cnblog\/204757\/"},"modified":"2025-05-29T16:29:45","modified_gmt":"2025-05-29T08:29:45","slug":"%e5%a6%82%e4%bd%95%e5%9c%a8-torch-tensorrt-%e4%b8%ad%e5%ae%9e%e7%8e%b0%e5%8a%a8%e6%80%81-batch-size%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/204757\/","title":{"rendered":"\u5982\u4f55\u5728 Torch-TensorRT \u4e2d\u5b9e\u73b0\u52a8\u6001 Batch Size\uff1f"},"content":{"rendered":"<p><b><\/b>     <\/p>\n<h1>\u5982\u4f55\u5728 Torch-TensorRT \u4e2d\u5b9e\u73b0\u52a8\u6001 Batch Size\uff1f<\/h1>\n<p>\u77e5\u8bc6\u70b9\u638c\u63e1\u4e86\uff0c\u8fd8\u9700\u8981\u4e0d\u65ad\u7ec3\u4e60\u624d\u80fd\u719f\u7ec3\u8fd0\u7528\u3002\u4e0b\u9762\u7ed9\u5927\u5bb6\u5e26\u6765\u4e00\u4e2a\u6587\u7ae0\u5f00\u53d1\u5b9e\u6218\uff0c\u624b\u628a\u624b\u6559\u5927\u5bb6\u5b66\u4e60\u300a\u5982\u4f55\u5728 Torch-TensorRT \u4e2d\u5b9e\u73b0\u52a8\u6001 Batch Size\uff1f\u300b\uff0c\u5728\u5b9e\u73b0\u529f\u80fd\u7684\u8fc7\u7a0b\u4e2d\u4e5f\u5e26\u5927\u5bb6\u91cd\u65b0\u6e29\u4e60\u76f8\u5173\u77e5\u8bc6\u70b9\uff0c\u6e29\u6545\u800c\u77e5\u65b0\uff0c\u56de\u5934\u770b\u770b\u8bf4\u4e0d\u5b9a\u53c8\u6709\u4e0d\u4e00\u6837\u7684\u611f\u609f\uff01<\/p>\n<p><img decoding=\"async\" src=\"https:\/\/www.17golang.com\/uploads\/20241115\/17316524286736eb4c0fea8.jpg\" class=\"aligncenter\"><\/p>\n<p><strong>\u5728 torch-tensorrt \u4e2d\u8bbe\u7f6e\u52a8\u6001 batch size<\/strong><\/p>\n<p>\u5728\u5c06 pytorch \u6a21\u578b\u8f6c\u6362\u4e3a tensorrt \u683c\u5f0f\u4ee5\u8fdb\u884c\u63a8\u7406\u65f6\uff0c\u6211\u4eec\u53ef\u80fd\u9700\u8981\u8bbe\u7f6e\u52a8\u6001 batch size \u6765\u9002\u5e94\u4e0d\u540c\u7684\u9884\u6d4b\u573a\u666f\u3002\u4f20\u7edf\u7684 compile() \u65b9\u5f0f\u65e0\u6cd5\u6ee1\u8db3\u8fd9\u4e00\u9700\u6c42\uff0c\u4ee5\u4e0b\u5c55\u793a\u5982\u4f55\u4f7f\u7528 input \u5bf9\u8c61\u8bbe\u7f6e\u52a8\u6001 batch size \u8303\u56f4\uff1a<\/p>\n<pre>from torch_tensorrt import Input\n\n# \u5b9a\u4e49\u8f93\u5165\u7ef4\u5ea6\nimage_channel = 3\nimage_size = 224\n\n# \u8bbe\u7f6e\u6700\u5c0f\u5f62\u72b6\u3001\u6700\u4f73\u5f62\u72b6\u548c\u6700\u5927\u5f62\u72b6\nmin_shape = [1, image_channel, image_size, image_size]\nopt_shape = [1, image_channel, image_size, image_size]\nmax_shape = [100, image_channel, image_size, image_size]\n\n# \u521b\u5efa Input \u5bf9\u8c61\ninputs = [\n    Input(min_shape, opt_shape, max_shape)\n]\n\n# \u7f16\u8bd1\u6a21\u578b\uff0c\u542f\u7528 fp16 \u7cbe\u5ea6\ntrt_ts_module = torch_tensorrt.compile(model, inputs, enabled_precisions={torch.float})<\/pre>\n<p>\u901a\u8fc7\u8bbe\u7f6e max_shape \u4e3a\u6240\u9700\u7684\u52a8\u6001 batch size \u4e0a\u9650\uff0c\u5373\u53ef\u5728\u7f16\u8bd1\u8fc7\u7a0b\u4e2d\u6307\u5b9a\u52a8\u6001 batch size \u8303\u56f4\u3002\u503c\u5f97\u6ce8\u610f\u7684\u662f\uff0c\u8fd9\u4e2a\u8303\u56f4\u5e94\u8be5\u6839\u636e\u786c\u4ef6\u8d44\u6e90\u548c\u663e\u5b58\u9650\u5236\u8fdb\u884c\u8c03\u6574\u3002<\/p>\n<p>\u4ee5\u4e0a\u5c31\u662f\u672c\u6587\u7684\u5168\u90e8\u5185\u5bb9\u4e86\uff0c\u662f\u5426\u6709\u987a\u5229\u5e2e\u52a9\u4f60\u89e3\u51b3\u95ee\u9898\uff1f\u82e5\u662f\u80fd\u7ed9\u4f60\u5e26\u6765\u5b66\u4e60\u4e0a\u7684\u5e2e\u52a9\uff0c\u8bf7\u5927\u5bb6\u591a\u591a\u652f\u6301\uff01\u66f4\u591a\u5173\u4e8e\u6587\u7ae0\u7684\u76f8\u5173\u77e5\u8bc6\uff0c\u4e5f\u53ef\u5173\u6ce8\u516c\u4f17\u53f7\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u5982\u4f55\u5728 Torch-Tensor&#46;&#46;&#46;<\/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-204757","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204757","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=204757"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204757\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=204757"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=204757"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=204757"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}