{"id":16495,"date":"2025-04-18T22:57:55","date_gmt":"2025-04-18T14:57:55","guid":{"rendered":"https:\/\/www.wsisp.com\/helps\/16495.html"},"modified":"2025-04-18T22:57:55","modified_gmt":"2025-04-18T14:57:55","slug":"%e4%bd%bf%e7%94%a8-tensorrt-%e5%92%8c-python-%e5%ae%9e%e7%8e%b0%e9%ab%98%e6%80%a7%e8%83%bd%e5%9b%be%e5%83%8f%e6%8e%a8%e7%90%86%e6%9c%8d%e5%8a%a1%e5%99%a8","status":"publish","type":"post","link":"https:\/\/www.wsisp.com\/helps\/16495.html","title":{"rendered":"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668"},"content":{"rendered":"<h2>\u5728\u73b0\u4ee3\u6df1\u5ea6\u5b66\u4e60\u548c\u8ba1\u7b97\u673a\u89c6\u89c9\u5e94\u7528\u4e2d\uff0c\u9ad8\u6027\u80fd\u63a8\u7406\u662f\u5173\u952e\u3002\u672c\u6587\u5c06\u4ecb\u7ecd\u5982\u4f55\u4f7f\u7528&nbsp;TensorRT&nbsp;\u548c&nbsp;Python&nbsp;\u6784\u5efa\u4e00\u4e2a\u9ad8\u6027\u80fd\u7684\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\u3002\u8be5\u670d\u52a1\u5668\u80fd\u591f\u63a5\u6536\u5ba2\u6237\u7aef\u53d1\u9001\u7684\u56fe\u50cf\u6570\u636e\uff0c\u4f7f\u7528 TensorRT \u8fdb\u884c\u63a8\u7406\uff0c\u5e76\u5c06\u7ed3\u679c\u8fd4\u56de\u7ed9\u5ba2\u6237\u7aef\u3002<\/h2>\n<h3>1. \u6982\u8ff0<\/h3>\n<h4>1.1 \u9879\u76ee\u76ee\u6807<\/h4>\n<ul>\n<li>\n<p>\u6784\u5efa\u4e00\u4e2a\u57fa\u4e8e TCP \u534f\u8bae\u7684\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\u3002<\/p>\n<\/li>\n<li>\n<p>\u4f7f\u7528 TensorRT \u52a0\u901f\u6df1\u5ea6\u5b66\u4e60\u6a21\u578b\u7684\u63a8\u7406\u3002<\/p>\n<\/li>\n<li>\n<p>\u652f\u6301\u591a\u5ba2\u6237\u7aef\u5e76\u53d1\u5904\u7406\u3002<\/p>\n<\/li>\n<\/ul>\n<h4>1.2 \u6280\u672f\u6808<\/h4>\n<ul>\n<li>\n<p>TensorRT\uff1aNVIDIA \u7684\u9ad8\u6027\u80fd\u6df1\u5ea6\u5b66\u4e60\u63a8\u7406\u5e93\u3002<\/p>\n<\/li>\n<li>\n<p>PyCUDA\uff1a\u7528\u4e8e\u5728 Python \u4e2d\u64cd\u4f5c CUDA\u3002<\/p>\n<\/li>\n<li>\n<p>OpenCV&nbsp;\u548c&nbsp;PIL\uff1a\u7528\u4e8e\u56fe\u50cf\u5904\u7406\u3002<\/p>\n<\/li>\n<li>\n<p>Socket\uff1a\u7528\u4e8e\u5b9e\u73b0\u7f51\u7edc\u901a\u4fe1\u3002<\/p>\n<\/li>\n<li>\n<p>\u591a\u8fdb\u7a0b\uff1a\u4f7f\u7528&nbsp;ProcessPoolExecutor&nbsp;\u5b9e\u73b0\u5e76\u53d1\u5904\u7406\u3002<\/p>\n<\/li>\n<\/ul>\n<hr>\n<h3>2. \u4ee3\u7801\u5b9e\u73b0<\/h3>\n<h4>2.1 \u4f9d\u8d56\u5e93<\/h4>\n<p>\u9996\u5148\uff0c\u786e\u4fdd\u5b89\u88c5\u4e86\u4ee5\u4e0b Python \u5e93\uff1a<\/p>\n<p>pip install numpy opencv-python pillow pycuda tensorrt torch<\/p>\n<h4>2.2 \u6838\u5fc3\u4ee3\u7801<\/h4>\n<h5>2.2.1 \u56fe\u50cf\u9884\u5904\u7406<\/h5>\n<p>def normalize(image: np.ndarray):<br \/>\n    \\&#8221;\\&#8221;\\&#8221;<br \/>\n    \u56fe\u50cf\u5f52\u4e00\u5316\u5904\u7406<br \/>\n    \\&#8221;\\&#8221;\\&#8221;<br \/>\n    resize_image = cv2.resize(image, (700, 800))<br \/>\n    image = resize_image.astype(np.float32)<br \/>\n    mean = (0.485, 0.456, 0.406)<br \/>\n    std = (0.229, 0.224, 0.225)<br \/>\n    image \/= 255.0<br \/>\n    image -= mean<br \/>\n    image \/= std<br \/>\n    return image <\/p>\n<h5>2.2.2 \u52a0\u8f7d TensorRT \u5f15\u64ce<\/h5>\n<p>def load_engine(engine_path):<br \/>\n    \\&#8221;\\&#8221;\\&#8221;<br \/>\n    \u52a0\u8f7d TensorRT \u5f15\u64ce<br \/>\n    \\&#8221;\\&#8221;\\&#8221;<br \/>\n    TRT_LOGGER = trt.Logger(trt.Logger.WARNING)<br \/>\n    with open(engine_path, \\&#8221;rb\\&#8221;) as f, trt.Runtime(TRT_LOGGER) as runtime:<br \/>\n        return runtime.deserialize_cuda_engine(f.read()) <\/p>\n<h5>2.2.3 TensorRT \u63a8\u7406<\/h5>\n<p>def ai_predict(input_data,engine,batch_size=1):<br \/>\n    with engine.create_execution_context() as context:<br \/>\n        stream = cuda.Stream()<br \/>\n        # \u5206\u914d\u8f93\u5165\u548c\u8f93\u51fa\u7684\u5185\u5b58<br \/>\n        print(input_data.dtype,input_data.nbytes,input_data.shape)<br \/>\n        h_output = cuda.pagelocked_empty((batch_size, 5), dtype=np.float32)  # \u5047\u8bbe\u8f93\u51fa\u662f 5 \u7c7b<br \/>\n        d_input = cuda.mem_alloc(input_data.nbytes)<br \/>\n        d<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u6587\u7ae0\u6d4f\u89c8\u9605\u8bfb792\u6b21\uff0c\u70b9\u8d5e19\u6b21\uff0c\u6536\u85cf18\u6b21\u3002\u5728\u73b0\u4ee3\u6df1\u5ea6\u5b66\u4e60\u548c\u8ba1\u7b97\u673a\u89c6\u89c9\u5e94\u7528\u4e2d\uff0c\u9ad8\u6027\u80fd\u63a8\u7406\u662f\u5173\u952e\u3002\u672c\u6587\u5c06\u4ecb\u7ecd\u5982\u4f55\u4f7f\u7528 TensorRT \u548c Python \u6784\u5efa\u4e00\u4e2a\u9ad8\u6027\u80fd\u7684\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\u3002\u8be5\u670d\u52a1\u5668\u80fd\u591f\u63a5\u6536\u5ba2\u6237\u7aef\u53d1\u9001\u7684\u56fe\u50cf\u6570\u636e\uff0c\u4f7f\u7528 TensorRT \u8fdb\u884c\u63a8\u7406\uff0c\u5e76\u5c06\u7ed3\u679c\u8fd4\u56de\u7ed9\u5ba2\u6237\u7aef\u3002_tensorrt python<\/p>\n","protected":false},"author":2,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[1],"tags":[81,152,86],"topic":[],"class_list":["post-16495","post","type-post","status-publish","format-standard","hentry","category-server","tag-python","tag-pytorch","tag-86"],"yoast_head":"<!-- This site is optimized with the Yoast SEO plugin v20.3 - https:\/\/yoast.com\/wordpress\/plugins\/seo\/ -->\n<title>\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3<\/title>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/www.wsisp.com\/helps\/16495.html\" \/>\n<meta property=\"og:locale\" content=\"zh_CN\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"og:description\" content=\"\u6587\u7ae0\u6d4f\u89c8\u9605\u8bfb792\u6b21\uff0c\u70b9\u8d5e19\u6b21\uff0c\u6536\u85cf18\u6b21\u3002\u5728\u73b0\u4ee3\u6df1\u5ea6\u5b66\u4e60\u548c\u8ba1\u7b97\u673a\u89c6\u89c9\u5e94\u7528\u4e2d\uff0c\u9ad8\u6027\u80fd\u63a8\u7406\u662f\u5173\u952e\u3002\u672c\u6587\u5c06\u4ecb\u7ecd\u5982\u4f55\u4f7f\u7528 TensorRT \u548c Python \u6784\u5efa\u4e00\u4e2a\u9ad8\u6027\u80fd\u7684\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\u3002\u8be5\u670d\u52a1\u5668\u80fd\u591f\u63a5\u6536\u5ba2\u6237\u7aef\u53d1\u9001\u7684\u56fe\u50cf\u6570\u636e\uff0c\u4f7f\u7528 TensorRT \u8fdb\u884c\u63a8\u7406\uff0c\u5e76\u5c06\u7ed3\u679c\u8fd4\u56de\u7ed9\u5ba2\u6237\u7aef\u3002_tensorrt python\" \/>\n<meta property=\"og:url\" content=\"https:\/\/www.wsisp.com\/helps\/16495.html\" \/>\n<meta property=\"og:site_name\" content=\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\" \/>\n<meta property=\"article:published_time\" content=\"2025-04-18T14:57:55+00:00\" \/>\n<meta name=\"author\" content=\"admin\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:label1\" content=\"\u4f5c\u8005\" \/>\n\t<meta name=\"twitter:data1\" content=\"admin\" \/>\n\t<meta name=\"twitter:label2\" content=\"\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4\" \/>\n\t<meta name=\"twitter:data2\" content=\"1 \u5206\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\/\/schema.org\",\"@graph\":[{\"@type\":\"WebPage\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/16495.html\",\"url\":\"https:\/\/www.wsisp.com\/helps\/16495.html\",\"name\":\"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"isPartOf\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\"},\"datePublished\":\"2025-04-18T14:57:55+00:00\",\"dateModified\":\"2025-04-18T14:57:55+00:00\",\"author\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\"},\"breadcrumb\":{\"@id\":\"https:\/\/www.wsisp.com\/helps\/16495.html#breadcrumb\"},\"inLanguage\":\"zh-Hans\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\/\/www.wsisp.com\/helps\/16495.html\"]}]},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/16495.html#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"\u9996\u9875\",\"item\":\"https:\/\/www.wsisp.com\/helps\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#website\",\"url\":\"https:\/\/www.wsisp.com\/helps\/\",\"name\":\"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3\",\"description\":\"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b\",\"potentialAction\":[{\"@type\":\"SearchAction\",\"target\":{\"@type\":\"EntryPoint\",\"urlTemplate\":\"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}\"},\"query-input\":\"required name=search_term_string\"}],\"inLanguage\":\"zh-Hans\"},{\"@type\":\"Person\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41\",\"name\":\"admin\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"zh-Hans\",\"@id\":\"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/\",\"url\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"contentUrl\":\"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery\",\"caption\":\"admin\"},\"sameAs\":[\"http:\/\/wp.wsisp.com\"],\"url\":\"https:\/\/www.wsisp.com\/helps\/author\/admin\"}]}<\/script>\n<!-- \/ Yoast SEO plugin. -->","yoast_head_json":{"title":"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/www.wsisp.com\/helps\/16495.html","og_locale":"zh_CN","og_type":"article","og_title":"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","og_description":"\u6587\u7ae0\u6d4f\u89c8\u9605\u8bfb792\u6b21\uff0c\u70b9\u8d5e19\u6b21\uff0c\u6536\u85cf18\u6b21\u3002\u5728\u73b0\u4ee3\u6df1\u5ea6\u5b66\u4e60\u548c\u8ba1\u7b97\u673a\u89c6\u89c9\u5e94\u7528\u4e2d\uff0c\u9ad8\u6027\u80fd\u63a8\u7406\u662f\u5173\u952e\u3002\u672c\u6587\u5c06\u4ecb\u7ecd\u5982\u4f55\u4f7f\u7528 TensorRT \u548c Python \u6784\u5efa\u4e00\u4e2a\u9ad8\u6027\u80fd\u7684\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668\u3002\u8be5\u670d\u52a1\u5668\u80fd\u591f\u63a5\u6536\u5ba2\u6237\u7aef\u53d1\u9001\u7684\u56fe\u50cf\u6570\u636e\uff0c\u4f7f\u7528 TensorRT \u8fdb\u884c\u63a8\u7406\uff0c\u5e76\u5c06\u7ed3\u679c\u8fd4\u56de\u7ed9\u5ba2\u6237\u7aef\u3002_tensorrt python","og_url":"https:\/\/www.wsisp.com\/helps\/16495.html","og_site_name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","article_published_time":"2025-04-18T14:57:55+00:00","author":"admin","twitter_card":"summary_large_image","twitter_misc":{"\u4f5c\u8005":"admin","\u9884\u8ba1\u9605\u8bfb\u65f6\u95f4":"1 \u5206"},"schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"WebPage","@id":"https:\/\/www.wsisp.com\/helps\/16495.html","url":"https:\/\/www.wsisp.com\/helps\/16495.html","name":"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668 - \u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","isPartOf":{"@id":"https:\/\/www.wsisp.com\/helps\/#website"},"datePublished":"2025-04-18T14:57:55+00:00","dateModified":"2025-04-18T14:57:55+00:00","author":{"@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41"},"breadcrumb":{"@id":"https:\/\/www.wsisp.com\/helps\/16495.html#breadcrumb"},"inLanguage":"zh-Hans","potentialAction":[{"@type":"ReadAction","target":["https:\/\/www.wsisp.com\/helps\/16495.html"]}]},{"@type":"BreadcrumbList","@id":"https:\/\/www.wsisp.com\/helps\/16495.html#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"\u9996\u9875","item":"https:\/\/www.wsisp.com\/helps"},{"@type":"ListItem","position":2,"name":"\u4f7f\u7528 TensorRT \u548c Python \u5b9e\u73b0\u9ad8\u6027\u80fd\u56fe\u50cf\u63a8\u7406\u670d\u52a1\u5668"}]},{"@type":"WebSite","@id":"https:\/\/www.wsisp.com\/helps\/#website","url":"https:\/\/www.wsisp.com\/helps\/","name":"\u7f51\u7855\u4e92\u8054\u5e2e\u52a9\u4e2d\u5fc3","description":"\u9999\u6e2f\u670d\u52a1\u5668_\u9999\u6e2f\u4e91\u670d\u52a1\u5668\u8d44\u8baf_\u670d\u52a1\u5668\u5e2e\u52a9\u6587\u6863_\u670d\u52a1\u5668\u6559\u7a0b","potentialAction":[{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https:\/\/www.wsisp.com\/helps\/?s={search_term_string}"},"query-input":"required name=search_term_string"}],"inLanguage":"zh-Hans"},{"@type":"Person","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/358e386c577a3ab51c4493330a20ad41","name":"admin","image":{"@type":"ImageObject","inLanguage":"zh-Hans","@id":"https:\/\/www.wsisp.com\/helps\/#\/schema\/person\/image\/","url":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","contentUrl":"https:\/\/gravatar.wp-china-yes.net\/avatar\/?s=96&d=mystery","caption":"admin"},"sameAs":["http:\/\/wp.wsisp.com"],"url":"https:\/\/www.wsisp.com\/helps\/author\/admin"}]}},"_links":{"self":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/16495","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/users\/2"}],"replies":[{"embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/comments?post=16495"}],"version-history":[{"count":0,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/posts\/16495\/revisions"}],"wp:attachment":[{"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/media?parent=16495"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/categories?post=16495"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/tags?post=16495"},{"taxonomy":"topic","embeddable":true,"href":"https:\/\/www.wsisp.com\/helps\/wp-json\/wp\/v2\/topic?post=16495"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}