{"id":204910,"date":"2025-05-29T16:33:26","date_gmt":"2025-05-29T08:33:26","guid":{"rendered":"https:\/\/server.hk\/cnblog\/204910\/"},"modified":"2025-05-29T16:33:26","modified_gmt":"2025-05-29T08:33:26","slug":"%e5%a6%82%e4%bd%95%e4%bd%bf%e7%94%a8-torch_tensorrt-%e8%ae%be%e7%bd%ae%e5%8a%a8%e6%80%81%e6%89%b9%e6%ac%a1%e5%a4%a7%e5%b0%8f%ef%bc%9f","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/204910\/","title":{"rendered":"\u5982\u4f55\u4f7f\u7528 torch_tensorrt \u8bbe\u7f6e\u52a8\u6001\u6279\u6b21\u5927\u5c0f\uff1f"},"content":{"rendered":"<p><b><\/b>     <\/p>\n<h1>\u5982\u4f55\u4f7f\u7528 torch_tensorrt \u8bbe\u7f6e\u52a8\u6001\u6279\u6b21\u5927\u5c0f\uff1f<\/h1>\n<p>\u4e00\u5206\u8015\u8018\uff0c\u4e00\u5206\u6536\u83b7\uff01\u65e2\u7136\u90fd\u6253\u5f00\u8fd9\u7bc7\uff0c\u5c31\u575a\u6301\u770b\u4e0b\u53bb\uff0c\u5b66\u4e0b\u53bb\u5427\uff01\u672c\u6587\u4e3b\u8981\u4f1a\u7ed9\u5927\u5bb6\u8bb2\u5230<span style=\"color: #FF6600;, Helvetica, Arial, sans-serif;font-size: 14px;background-color: #FFFFFF\"><\/span>\u7b49\u7b49\u77e5\u8bc6\u70b9\uff0c\u5982\u679c\u5927\u5bb6\u5bf9\u672c\u6587\u6709\u597d\u7684\u5efa\u8bae\u6216\u8005\u770b\u5230\u6709\u4e0d\u8db3\u4e4b\u5904\uff0c\u975e\u5e38\u6b22\u8fce\u5927\u5bb6\u79ef\u6781\u63d0\u51fa\uff01\u5728\u540e\u7eed\u6587\u7ae0\u6211\u4f1a\u7ee7\u7eed\u66f4\u65b0<span style=\"color: #FF6600;, Helvetica, Arial, sans-serif;font-size: 14px;background-color: #FFFFFF\">\u6587\u7ae0<\/span>\u76f8\u5173\u7684\u5185\u5bb9\uff0c\u5e0c\u671b\u5bf9\u5927\u5bb6\u90fd\u6709\u6240\u5e2e\u52a9\uff01<\/p>\n<p><img decoding=\"async\" src=\"https:\/\/www.17golang.com\/uploads\/20241119\/1731985260673bff6caeea3.jpg\" class=\"aligncenter\"><\/p>\n<p><strong>\u5728 torch_tensorrt \u4e2d\u8bbe\u7f6e\u52a8\u6001\u6279\u6b21\u5927\u5c0f<\/strong><\/p>\n<p>\u5728\u4f7f\u7528 pytorch \u6a21\u578b\u4e0e torch-tensorrt \u8fdb\u884c\u63a8\u7406\u65f6\uff0c\u9700\u8981\u6307\u5b9a\u6279\u6b21\u5927\u5c0f\u3002\u800c\u5bf9\u4e8e\u63a8\u7406\u573a\u666f\uff0c\u6279\u6b21\u5927\u5c0f\u5f80\u5f80\u662f\u52a8\u6001\u53d8\u5316\u7684\uff0c\u56e0\u6b64\u9700\u8981\u5728\u5c06\u6a21\u578b\u8f6c\u6362\u4e3a tensorrt \u683c\u5f0f\u65f6\u8bbe\u7f6e\u52a8\u6001\u6279\u6b21\u5927\u5c0f\u3002<\/p>\n<p>\u5728 torch_tensorrt \u7684 compile \u51fd\u6570\u4e2d\uff0c\u8f93\u5165\u53c2\u6570 inputs \u5b9a\u4e49\u4e86\u6a21\u578b\u63a8\u7406\u65f6\u7684\u5f62\u72b6\u8303\u56f4\u3002\u8981\u8bbe\u7f6e\u52a8\u6001\u6279\u6b21\u5927\u5c0f\uff0c\u9700\u8981\u6307\u5b9a min_shape\u3001opt_shape \u548c max_shape \u4e09\u4e2a\u53c2\u6570\uff1a<\/p>\n<ul>\n<li>min_shape: \u63a8\u7406\u7684\u6700\u5c0f\u6279\u6b21\u5927\u5c0f<\/li>\n<li>opt_shape: \u63a8\u7406\u7684\u63a8\u8350\u6279\u6b21\u5927\u5c0f<\/li>\n<li>max_shape: \u63a8\u7406\u7684\u6700\u5927\u6279\u6b21\u5927\u5c0f<\/li>\n<\/ul>\n<p>\u4f8b\u5982\uff0c\u8981\u5c06\u6279\u6b21\u5927\u5c0f\u8bbe\u7f6e\u4e3a 1 \u5230 100 \u4e4b\u95f4\u7684\u52a8\u6001\u5927\u5c0f\uff0c\u53ef\u4ee5\u8fd9\u6837\u8bbe\u7f6e\uff1a<\/p>\n<pre>inputs = [\n    torch_tensorrt.Input(\n        min_shape=[1, image_channel, image_size, image_size],\n        opt_shape=[1, image_channel, image_size, image_size],\n        max_shape=[100, image_channel, image_size, image_size],  # \u8bbe\u7f6e\u6700\u5927\u6279\u6b21\u5927\u5c0f\u4e3a 100\n        device=device\n    )\n]<\/pre>\n<p>\u8bf7\u6ce8\u610f\uff0c\u52a8\u6001\u6279\u6b21\u5927\u5c0f\u7684\u8303\u56f4\u5e94\u6839\u636e\u786c\u4ef6\u548c\u663e\u5b58\u9650\u5236\u8fdb\u884c\u6743\u8861\uff0c\u4ee5\u786e\u4fdd\u6a21\u578b\u80fd\u591f\u9ad8\u6548\u8fd0\u884c\u3002<\/p>\n<p>\u4eca\u5929\u5e26\u5927\u5bb6\u4e86\u89e3\u4e86\u7684\u76f8\u5173\u77e5\u8bc6\uff0c\u5e0c\u671b\u5bf9\u4f60\u6709\u6240\u5e2e\u52a9\uff1b\u5173\u4e8e\u6587\u7ae0\u7684\u6280\u672f\u77e5\u8bc6\u6211\u4eec\u4f1a\u4e00\u70b9\u70b9\u6df1\u5165\u4ecb\u7ecd\uff0c\u6b22\u8fce\u5927\u5bb6\u5173\u6ce8\u516c\u4f17\u53f7\uff0c\u4e00\u8d77\u5b66\u4e60\u7f16\u7a0b~<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u5982\u4f55\u4f7f\u7528 torch_tenso&#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-204910","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204910","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=204910"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/204910\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=204910"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=204910"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=204910"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}