TensorRT安裝及使用

2021-10-04 03:51:22 字數 548 閱讀 5352

如果出錯了看是不是這個問題:

tensorrt官方安裝、公升級、解除安裝指南:

下邊的內容我是學習b站上一位up主的教程,自己順便記錄一下學習的內容。

import tensorrt as trt
$convert-to-uff frozen_inference_graph.pb
>>>model_file = './data/mnist/mnsit.uff'
with builder = trt.builder(trt_logger) as builder,builder.create_network() as network,trt.uffparser() as parser:

parser.register_input('placeholder',(1,28,28))

parser.register_output('fc2/relu')

parser.parse(model_file,network)

TensorRT安裝與使用(一)

參考安裝教程請直接轉到3.1 tar安裝 nvidia tensorrt 的核心是乙個c 庫,有助於對 nvidia 圖形處理單元 gpu 進行高效能推理。tensorrt 採用經過訓練的網路,該網路由網路定義和一組經過訓練的引數組成,並生成高度優化的執行時引擎,為該網路執行推理。tensorrt通...

tensorrt安裝教程及失敗情況

tensorrt安裝教程 在進入 tensorrt samples資料夾執行make報錯 錯誤 dkpg query 沒有找到與cuda cudart 0 9 相匹配得軟體包 解決辦法 在 tensorrt samples 資料夾中有個 makefile.config 檔案,裡面第4行 cuda v...

LINUX安裝TensorRT及特別注意事項

柳鯤鵬 這個在tensorrt的python目錄下,可以看到包含有cpxx的whl檔案,就是對用的python版本號。解壓tar xvf tensorrt 4.0.1.6.ubuntu 16.04.4.x86 64 gnu.cuda 9.2.cudnn7.1.tar.gzcd python sudo...